輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。
例如,給出
前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]
返回如下的二叉樹:
3/ \
9 20
/ \
15 7
3 9 【 20 15 7 】
根節點 左子樹 右子樹
9 3 【 15 20 7 】
左子樹 根節點 右子樹
根據前序遍歷,可以確定根節點;根據中序遍歷,可以確定左子樹節點和右子樹節點,同時可以確定子樹節點的前序遍歷和中序遍歷結果,因此根據節點使用遞迴方法建立子樹。
1#定義二叉樹
2class
treenode:
3def
__init__(self, x): #
self相當於類的例項
4 self.val =x
5 self.left =none
6 self.right =none78
9class
solution:
10def buildtree(self, preorder: list[int], inorder: list[int]) ->treenode:
11if
not preorder or
not inorder or len(preorder) != len(inorder): #
考慮異常條件
12return
none
13 root =preorder[0]
14 rootnode = treenode(root) #
建立根節點
15 pos = inorder.index(root) #
確定根節點在中序遍歷中的索引位置
1617 preorderleft = preorder[1:pos+1] #
前序遍歷的左子樹
18 preorderright = preorder[pos+1:] #
前序遍歷的右子樹
1920 inorderleft = inorder[:pos] #
中序遍歷的左子樹
21 inorderright = inorder[pos+1:] #
中序遍歷的右子樹
2223 leftnode = self.buildtree(preorderleft, inorderleft) #
遞迴到左子樹的根節點
24 rightnode = self.buildtree(preorderright, inorderright) #
遞迴到右子樹的根節點
25 rootnode.left =leftnode
26 rootnode.right =rightnode
27return rootnode
1struct
treenode 6};
78class
solution
2223 vector preorderleft; //
前序遍歷的左子樹
24 vector preorderright; //
前序遍歷的右子樹
25for (int i = 1; i <= pos; i++)
26preorderleft.push_back(preorder[i]);
27for (int i = pos + 1; i < preorder.size(); i++)
28preorderright.push_back(preorder[i]);
2930 vector inorderleft; //
中序遍歷的左子樹
31 vector inorderright; //
中序遍歷的右子樹
32for (int i = 0; i < pos; i++)
33inorderleft.push_back(inorder[i]);
34for (int i = pos + 1; i < inorder.size(); i++)
35inorderright.push_back(inorder[i]);
3637 treenode *leftnode = buildtree(preorderleft, inorderleft); //
遞迴到左子樹的根節點
38 treenode *rightnode = buildtree(preorderright, inorderright); //
遞迴到右子樹的根節點
39 rootnode->left =leftnode;
40 rootnode->right =rightnode;
4142
return
rootnode;43}
44 };
二叉樹 重建二叉樹
問題 給定二叉樹的前序遍歷結果和中序遍歷結果,恢復出原二叉樹。假設二叉樹中的元素都不重複,給定二叉樹的前序遍歷序列,二叉樹的中序遍歷序列。看到此題,我首先想到的是尋找根節點,由前序遍歷序列可以看出根節點為1,此時通過中序遍歷可以看出來4,7,2在根節點的左子樹,5,3,8,6在樹的右節點。此時我們可...
二叉樹 重建二叉樹
題目給定兩個陣列,乙個是前序遍歷陣列 preorder 乙個是中序遍歷陣列 inorder 要求輸出還原二叉樹 核心在於我們要理解前序和中序便利的特點 前序遍歷 根節點 左節點 右節點 中序遍歷 左節點 根節點 右節點 所以我們從二叉樹的根節點開始重構 也就是preorder的第乙個值 同時用乙個m...
二叉樹重建
摘自劉汝佳的 演算法競賽入門經典 preorder t t 的根結點 preorder t 的左子樹 preorder t 的右子樹 inorder t inorder t 的左子樹 t 的根結點 inorder t 的右子樹 postorder t postorder t 的左子樹 postord...