mirror of
https://github.com/krahets/hello-algo.git
synced 2025-07-06 22:34:18 +08:00
translation: Add Python and Java code for EN version (#1345)
* Add the intial translation of code of all the languages * test * revert * Remove * Add Python and Java code for EN version
This commit is contained in:
81
en/codes/python/modules/print_util.py
Normal file
81
en/codes/python/modules/print_util.py
Normal file
@ -0,0 +1,81 @@
|
||||
"""
|
||||
File: print_util.py
|
||||
Created Time: 2021-12-11
|
||||
Author: krahets (krahets@163.com), msk397 (machangxinq@gmail.com)
|
||||
"""
|
||||
|
||||
from .tree_node import TreeNode, list_to_tree
|
||||
from .list_node import ListNode, linked_list_to_list
|
||||
|
||||
|
||||
def print_matrix(mat: list[list[int]]):
|
||||
"""Print matrix"""
|
||||
s = []
|
||||
for arr in mat:
|
||||
s.append(" " + str(arr))
|
||||
print("[\n" + ",\n".join(s) + "\n]")
|
||||
|
||||
|
||||
def print_linked_list(head: ListNode | None):
|
||||
"""Print linked list"""
|
||||
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
|
||||
):
|
||||
"""
|
||||
Print binary tree
|
||||
This tree printer is borrowed from 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):
|
||||
"""Print dictionary"""
|
||||
for key, value in hmap.items():
|
||||
print(key, "->", value)
|
||||
|
||||
|
||||
def print_heap(heap: list[int]):
|
||||
"""Print heap"""
|
||||
print("Array representation of the heap:", heap)
|
||||
print("Tree representation of the heap:")
|
||||
root: TreeNode | None = list_to_tree(heap)
|
||||
print_tree(root)
|
Reference in New Issue
Block a user