refactor: Follow the PEP 585 Typing standard (#439)

* Follow the PEP 585 Typing standard

* Update list.py
This commit is contained in:
Yudong Jin
2023-03-23 18:51:56 +08:00
committed by GitHub
parent f4e01ea32e
commit 8918ec9079
43 changed files with 256 additions and 342 deletions

View File

@ -7,33 +7,31 @@ Author: Krahets (krahets@163.com), msk397 (machangxinq@gmail.com)
from .binary_tree import TreeNode, list_to_tree
from .linked_list import ListNode, linked_list_to_list
from typing import List, Optional, Dict
def print_matrix(mat: List[List[int]]) -> None:
def print_matrix(mat: list[list[int]]) -> None:
""" Print a matrix """
s: List[str] = []
s: list[str] = []
for arr in mat:
s.append(' ' + str(arr))
print('[\n' + ',\n'.join(s) + '\n]')
def print_linked_list(head: Optional[ListNode]) -> None:
def print_linked_list(head: ListNode | None) -> None:
""" Print a linked list """
arr: List[int] = linked_list_to_list(head)
arr: list[int] = linked_list_to_list(head)
print(' -> '.join([str(a) for a in arr]))
class Trunk:
def __init__(self, prev: Optional['Trunk'] = None, string: Optional[str] = None) -> None:
self.prev: Optional[Trunk] = prev
self.str: Optional[str] = string
def __init__(self, prev, string: str | None = None) -> None:
self.prev = prev
self.str = string
def show_trunks(p: Optional[Trunk]) -> None:
def show_trunks(p: Trunk | None) -> None:
if p is None:
return
show_trunks(p.prev)
print(p.str, end='')
def print_tree(root: Optional[TreeNode], prev: Optional[Trunk] = None, is_left: bool = False) -> None:
def print_tree(root: TreeNode | None, prev: Trunk | None = None, is_left: bool = False) -> None:
"""
Print a binary tree
This tree printer is borrowed from TECHIE DELIGHT
@ -62,14 +60,14 @@ def print_tree(root: Optional[TreeNode], prev: Optional[Trunk] = None, is_left:
trunk.str = ' |'
print_tree(root.left, trunk, False)
def print_dict(mapp: Dict) -> None:
def print_dict(mapp: dict) -> None:
""" Print a dict """
for key, value in mapp.items():
print(key, '->', value)
def print_heap(heap: List[int]) -> None:
def print_heap(heap: list[int]) -> None:
""" Print a heap both in array and tree representations """
print("堆的数组表示:", heap)
print("堆的树状表示:")
root: Optional[TreeNode] = list_to_tree(heap)
root: TreeNode | None = list_to_tree(heap)
print_tree(root)