剑指offer:面试题07. 重建二叉树
生活随笔
收集整理的這篇文章主要介紹了
剑指offer:面试题07. 重建二叉树
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
題目:?重建二叉樹
輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果,請重建該二叉樹。假設(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)個數(shù) <= 5000
解題:
- 前序遍歷:先訪問根結(jié)點(diǎn),再訪問左子結(jié)點(diǎn),最后訪問右子結(jié)點(diǎn)。
- 中序遍歷:先訪問左子結(jié)點(diǎn),再訪問根結(jié)點(diǎn),最后訪問右子結(jié)點(diǎn)。
- 后序遍歷:先訪問左子結(jié)點(diǎn),再訪問右子結(jié)點(diǎn),最后訪問根結(jié)點(diǎn)。
/*** 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 {
public:unordered_map<int, int> mp;vector<int> Pre, In;TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {Pre = preorder, In = inorder;int n = Pre.size();for(int i = 0; i < n; i++) mp[In[i]] = i;return dfs(0, n-1, 0, n-1);}TreeNode* dfs(int pL, int pR, int inL, int inR){if(pL > pR) return nullptr;auto root = new TreeNode(Pre[pL]);int k = mp[root->val];int left = k - inL;root->left = dfs(pL+1, pL + left, inL, k - 1);root->right = dfs(pL + left + 1, pR, k + 1, inR);return root;}
};
?
總結(jié)
以上是生活随笔為你收集整理的剑指offer:面试题07. 重建二叉树的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 剑指offer:面试题06. 从尾到头打
- 下一篇: 剑指offer:面试题09. 用两个栈实