重建二叉樹(Python and C 解法)

2022-08-24 19:36:12 字數 2829 閱讀 4256

輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。

例如,給出

前序遍歷 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

1

struct

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...