Fine tune

fnjerry · Dec 27, 2022 · f9cc3a5 · f9cc3a5
1 parent 449258f
commit f9cc3a5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 12 deletions.
diff --git a/codes/python/include/binary_tree.py b/codes/python/include/binary_tree.py
@@ -9,13 +9,13 @@
 class TreeNode:
     """Definition for a binary tree node
     """    
-    def __init__(self, val=None, left=None, right=None):
-        self.val = val  # 结点值
-        self.height = 0  # 结点高度, avl 树会用到
-        self.left = left  # 左子结点引用
+    def __init__(self, val=0, left=None, right=None):
+        self.val = val      # 结点值
+        self.height = 0     # 结点高度
+        self.left = left    # 左子结点引用
         self.right = right  # 右子结点引用
 
-    def __str__(self):  # 直接print时会好看一点
+    def __str__(self):
         val = self.val
         left_node_val = self.left.val if self.left else None
         right_node_val = self.right.val if self.right else None

diff --git a/docs/chapter_tree/binary_tree.md b/docs/chapter_tree/binary_tree.md
@@ -39,13 +39,6 @@ comments: true
             self.val = val      # 结点值
             self.left = left    # 左子结点指针
             self.right = right  # 右子结点指针
-
-        def __str__(self):
-            val = self.val
-            left_node_val = self.left.val if self.left else None
-            right_node_val = self.right.val if self.right else None
-            return "<TreeNode: {}, leftTreeNode: {}, rightTreeNode: {}>".format(val, left_node_val, right_node_val)
-
     ```
 
 === "Go"