前序数组的结构为:根—左子树所有结点—右子树所有结点
中序数组的结构为:左子树所有结点—根—右子树所有结点
确定根在中序数组的位置即可确定左右子树的大小,从而从前序数组找到左儿子结点和右儿子结点的值。之后,分别以左儿子和右儿子结点为根,进行相同操作。
时间复杂度:O(N^2)(最坏的情况为所有节点均为左子树,每次find操作都要遍历未放进二叉树的所有值,总操作数为
n
(
n
+
1
)
2
\frac{n(n+1)}{2}
2n(n+1))
C++代码:
class Solution {
public:
TreeNode * buildTree(vector<int>& preorder, vector<int>& inorder) {
if (preorder.empty())
return {};
return buildSubTree(preorder, 0, inorder, 0, inorder.size() - 1);
}
TreeNode* buildSubTree(vector<int>& preorder, int stP, vector<int>& inorder, int stI, int enI)
{
TreeNode* root = new TreeNode(preorder[stP]);
auto it = find(inorder.begin() + stI, inorder.begin() + enI + 1, *(preorder.begin() + stP));
int leftLen = it - inorder.begin() - stI, rightLen = inorder.begin() + enI - it;
if (leftLen > 0)
root->left = buildSubTree(preorder, stP + 1, inorder, stI, stI + leftLen - 1);
if (rightLen > 0)
root->right = buildSubTree(preorder, stP + leftLen + 1,
inorder, it - inorder.begin() + 1, enI);
return root;
}
};