二叉樹查找后繼節點(即中序遍歷情況下的這個節點的下一個) Python實現

本文轉載自查看原文 2018-06-04 08:17 920 算法

1.若節點類型沒有parent屬性，采用中序遍歷方式獲取后繼節點

 1 def getSuccessorNode(head, node):
 2     if (not node) or (not head):
 3         return None
 4     stack = []
 5     flag = False
 6     while head or len(stack) > 0:
 7         if head:
 8             stack.append(head)
 9             head = head.left
10         else:
11             head = stack.pop()
12             if flag:
13                 return head
14             if head == node:            # 若找到當前節點，則下一個彈出的節點即為后繼節點
15                 flag = True
16             head = head.right
17     return None

2.若節點存在parent屬性即

 1 class TreeNode:
 2     def __init__(self, x=0):
 3         self.val = x
 4         self.parent = None
 5         self.left = None
 6         self.right = None
 7 
 8 
 9 def getSuccessorNode(node):
10     if not node :
11         return None
12     if node.right:               # 如果當前節點有右子樹，則返回右子樹的最左邊節點
13         node = node.right
14         while node.left:
15             node = node.left
16         return node
17     else:                        # 沒有右子樹   則向上找尋父節點，直到為父節點的左子樹，返回父節點，否則返回空
18         par = node.parent
19         while not par and par.left != node:
20             node = par
21             par = par.parent
22         return par

免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。

猜您在找 二叉樹中序遍歷的下一個節點二叉樹搜索樹中序遍歷下的前驅節點與后繼節點二叉樹——在二叉樹中找到一個節點的后繼節點二叉樹 Java 實現前序遍歷中序遍歷后序遍歷層級遍歷獲取葉節點寬度，高度，隊列實現二叉樹遍歷求二叉樹的最大距離二叉樹的先序，中序，后序遞歸遍歷，以及計算節點的數目二叉樹查找節點 Leetcode練習(Python)：樹類：第102題：二叉樹的層序遍歷：給你一個二叉樹，請你返回其按層序遍歷得到的節點值。（即逐層地，從左到右訪問所有節點）。二叉樹中刪除一個節點用c語言實現前序創建二叉樹（遞歸），分別用前序，中序，后序遍歷，以及分別輸出節點個數和葉子節點個數由前序遍歷和中序遍歷構建二叉樹-Python