mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-02 03:05:24 +08:00
Yudong Jin
2f561d3d51
* feat(ko): add the complete Korean online edition * Use UTF-8 for Python CI output on Windows * Remove print-book promotion from the Korean site * fix(ko): preserve emphasis before Korean particles
82 lines
1.9 KiB
Python
82 lines
1.9 KiB
Python
"""
|
|
File: print_util.py
|
|
Created Time: 2021-12-11
|
|
Author: krahets ([email protected]), msk397 ([email protected])
|
|
"""
|
|
|
|
from .tree_node import TreeNode, list_to_tree
|
|
from .list_node import ListNode, linked_list_to_list
|
|
|
|
|
|
def print_matrix(mat: list[list[int]]):
|
|
"""행렬 출력"""
|
|
s = []
|
|
for arr in mat:
|
|
s.append(" " + str(arr))
|
|
print("[\n" + ",\n".join(s) + "\n]")
|
|
|
|
|
|
def print_linked_list(head: ListNode | None):
|
|
"""연결 리스트 출력"""
|
|
arr: list[int] = linked_list_to_list(head)
|
|
print(" -> ".join([str(a) for a in arr]))
|
|
|
|
|
|
class Trunk:
|
|
def __init__(self, prev, string: str | None = None):
|
|
self.prev = prev
|
|
self.str = string
|
|
|
|
|
|
def show_trunks(p: Trunk | None):
|
|
if p is None:
|
|
return
|
|
show_trunks(p.prev)
|
|
print(p.str, end="")
|
|
|
|
|
|
def print_tree(
|
|
root: TreeNode | None, prev: Trunk | None = None, is_right: bool = False
|
|
):
|
|
"""
|
|
이진 트리 출력
|
|
이 트리 출력 코드는 TECHIE DELIGHT에서 가져왔다
|
|
https://www.techiedelight.com/c-program-print-binary-tree/
|
|
"""
|
|
if root is None:
|
|
return
|
|
|
|
prev_str = " "
|
|
trunk = Trunk(prev, prev_str)
|
|
print_tree(root.right, trunk, True)
|
|
|
|
if prev is None:
|
|
trunk.str = "———"
|
|
elif is_right:
|
|
trunk.str = "/———"
|
|
prev_str = " |"
|
|
else:
|
|
trunk.str = "\———"
|
|
prev.str = prev_str
|
|
|
|
show_trunks(trunk)
|
|
print(" " + str(root.val))
|
|
if prev:
|
|
prev.str = prev_str
|
|
trunk.str = " |"
|
|
print_tree(root.left, trunk, False)
|
|
|
|
|
|
def print_dict(hmap: dict):
|
|
"""딕셔너리 출력"""
|
|
for key, value in hmap.items():
|
|
print(key, "->", value)
|
|
|
|
|
|
def print_heap(heap: list[int]):
|
|
"""힙 출력"""
|
|
print("힙의 배열 표현:", heap)
|
|
print("힙의 트리 표현:")
|
|
root: TreeNode | None = list_to_tree(heap)
|
|
print_tree(root)
|