Leetcode105.Construct_Binary_Tree_From_Preorder_And_Inorder_Traversal

前序数组的结构为:根—左子树所有结点—右子树所有结点
中序数组的结构为:左子树所有结点—根—右子树所有结点
确定根在中序数组的位置即可确定左右子树的大小,从而从前序数组找到左儿子结点和右儿子结点的值。之后,分别以左儿子和右儿子结点为根,进行相同操作。
时间复杂度:O(N^2)(最坏的情况为所有节点均为左子树,每次find操作都要遍历未放进二叉树的所有值,总操作数为 n ( n + 1 ) 2 \frac{n(n+1)}{2}
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;
	}
};

猜你喜欢

转载自blog.csdn.net/qq_42263831/article/details/83621502
今日推荐