重建二叉树

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

IDEA

着重说的点是find(),vector<int> vec1=vector<int>(vec.begin(),vec.begin()+a)的应用

//Definition for binary tree
struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x):val(x),left(NULL),right(NULL){}
};

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
    	if(pre.size()==0){
    		return NULL;
		}else if(pre.size()==1){
			return new TreeNode(pre[0]);
		}
		TreeNode* root=new TreeNode(pre[0]);
		int pos=std::find(vin.begin(),vin.end(),pre[0])-vin.begin();
		vector<int> l_pre=vector<int>(pre.begin()+1,pre.begin()+1+pos);
		vector<int> l_in=vector<int>(vin.begin(),vin.begin()+pos);
		vector<int> r_pre=vector<int>(pre.begin()+1+pos,pre.end());
		vector<int> r_in=vector<int>(vin.begin()+pos+1,vin.end());
		root->left=reConstructBinaryTree(l_pre,l_in);
		root->right=reConstructBinaryTree(r_pre,r_in);
		return root;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值