劍指 Offer 07. 重建二叉樹

輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果，請(qǐng)重建該二叉樹。假設(shè)輸入的前序遍歷和中序遍歷的結(jié)果中都不含重復(fù)的數(shù)字。

例如，給出

前序遍歷 preorder =?[3,9,20,15,7] 中序遍歷 inorder = [9,3,15,20,7]

返回如下的二叉樹：

3/ \9 20/ \15 7

限制：

0 <= 節(jié)點(diǎn)個(gè)數(shù) <= 5000

算法：

遞歸

C++代碼

/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode(int x) : val(x), left(NULL), right(NULL) {}* };*/ class Solution {unordered_map<int, int> index; public:TreeNode* Rebuildtree(vector<int>& preorder, vector<int>& inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right){if(preorder_left > preorder_right)return NULL;//前序遍歷的第一個(gè)節(jié)點(diǎn)就是根節(jié)點(diǎn)(preorder_left)int preorder_root = preorder_left;//根節(jié)點(diǎn)在中序遍歷出現(xiàn)的位置int inorder_root = index[preorder[preorder_root]];//創(chuàng)建根節(jié)點(diǎn)TreeNode* root = new TreeNode(preorder[preorder_root]);//得到左子樹中的節(jié)點(diǎn)數(shù)目int size_left_subtree = inorder_root - inorder_left;//遞歸構(gòu)造左子樹 并連接到根節(jié)點(diǎn)//前序遍歷的「從 左邊界+1 開始的 size_left_subtree」個(gè)元素就對(duì)應(yīng)了中序遍歷中「從 左邊界 開始到 根節(jié)點(diǎn)定位-1」的元素root->left = Rebuildtree(preorder, inorder, preorder_left+1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1);root->right = Rebuildtree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right);return root;}TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {int len = preorder.size();//構(gòu)建哈希表 方便后續(xù)直接在中序遍歷中找到根節(jié)點(diǎn)的位置for(int i = 0; i < len; i++){index[inorder[i]] = i;} //提供前序遍歷和中序遍歷的值及其范圍return Rebuildtree(preorder, inorder, 0, len - 1, 0, len - 1);} };

?

總結(jié)

以上是生活随笔為你收集整理的根据二叉树前序遍历和中序遍历重建二叉树的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。