【construct-binary-tree-from-preorder-and-inorder-traversal】

搬砖小能手lp

于 2017-09-09 09:12:43 发布

阅读量324

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode记录

本文链接：https://blog.csdn.net/lplp90908/article/details/77894522

leetcode记录专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种使用前序遍历和中序遍历来重构二叉树的方法。采用深度优先搜索（DFS）策略实现，通过递归找到每个子树的根节点，并进一步构建整棵树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:

You may assume that duplicates do not exist in the tree.

前序和中序重构二叉树；

这里用dfs的思想：

class Solution {
public:
    TreeNode *dfs(vector<int> &preorder,int preStart,int preEnd,vector<int> &inorder,int inStart,int inEnd)
	{
    	if(preStart > preEnd)
        	return NULL;
    	TreeNode *root = new TreeNode(preorder[preStart]);
    	int middle;
    	for(middle=inStart;middle<=inEnd;middle++)
        	if(inorder[middle] == root->val)
            	break;
    	int leftLen = middle-inStart;
    	root->left = dfs(preorder,preStart+1,preStart+leftLen,inorder,inStart,middle-1);
    	root->right = dfs(preorder,preStart+leftLen+1,preEnd,inorder,middle+1,inEnd);
    	return root;
	}
    
	TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)
	{
    	int pre = preorder.size(),in = inorder.size();
    	if(pre==0 || in==0 || pre != in)
        	return NULL;
    	return dfs(preorder,0,pre-1,inorder,0,in-1);
	}
};