题目地址:
https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
给定一棵树的先序和中序遍历,返回这棵树。题目保证这棵树的每个数两两不同。
思路是用递归。由先序遍历可以确定这棵树的根,然后再在中序遍历里去找这个树根,这样在中序遍历的序列里,树根左边就是左子树的所有节点,右边就是右子树的所有节点,也知道了左右子树的size是多少。这样就可以再在先序遍历的树根后数若干数作为左子树,再数若干数作为右子树,再递归求解即可。为了更快的在中序遍历里找到树根,可以用一个哈希表记录每个数在中序遍历数组的下标。代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
* right(right) {}
* };
*/
class Solution {
public:
TreeNode* buildTree(vector<int>& pre, vector<int>& in) {
unordered_map<int, int> mp;
int n = in.size();
for (int i = 0; i < n; i++) mp[in[i]] = i;
return dfs(pre, 0, n - 1, in, 0, n - 1, mp);
}
TreeNode* dfs(auto& pre, int b1, int e1, auto& in, int b2, int e2, auto& mp) {
if (b1 > e1) return nullptr;
auto root = new TreeNode(pre[b1]);
int idx = mp[root->val], sz_left = idx - b2;
root->left = dfs(pre, b1 + 1, b1 + sz_left, in, b2, idx - 1, mp);
root->right = dfs(pre, b1 + sz_left + 1, e1, in, idx + 1, e2, mp);
return root;
}
};
时空复杂度 O ( n ) O(n) O(n)。
注解:
关于先序遍历和中序遍历可以确定一棵二叉树的结论,可以用数学归纳法证明。设树节点个数为
0
0
0或
1
1
1,则结论成立。假设对于节点个数小于等于
n
n
n的树结论成立。当节点个数为
n
+
1
n+1
n+1时,由上面的算法可以先找到树根,接着由于左右子树节点数严格小于
n
+
1
n+1
n+1,由归纳假设,左右子树唯一确定,所以整棵树也唯一确定。