leetcode105. 从前序与中序遍历序列构造二叉树

最新推荐文章于 2022-12-13 21:12:36 发布

一条特立独行的狗、

最新推荐文章于 2022-12-13 21:12:36 发布

阅读量253

点赞数

分类专栏： Leetcode数组文章标签： c++ leetcode

本文链接：https://blog.csdn.net/weixin_43498985/article/details/103962961

版权

Leetcode数组专栏收录该内容

68 篇文章 0 订阅

订阅专栏

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:

你可以假设树中没有重复的元素。

例如，给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：
在这里插入图片描述

class Solution {
public:    
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)  {       	
	
	if( preorder.empty() || inorder.empty() )           
		return NULL;        
	int nRootVal = preorder[0];        
	TreeNode * pRoot = new TreeNode(nRootVal);           
	if(!pRoot)            
		return NULL;        
	int nIdx = -1;        
	for(int i=0;i<inorder.size();i++)        
	{            
		if(inorder[i] == nRootVal)            
		{                
			nIdx = i;               
			break;            
		}        
	}
        if(nIdx != -1)        
	{           
		vector<int> vecLeftPre(preorder.begin()+1,preorder.begin()+nIdx+1);            						
		vector<int> vecLeftIn(inorder.begin(),inorder.begin()+nIdx);
		vector<int> vecRightPre(preorder.begin()+nIdx+1,preorder.end());            
		vector<int> vecRightIn(inorder.begin()+nIdx+1,inorder.end())；

		pRoot->left = buildTree(vecLeftPre,vecLeftIn);            
		pRoot->right = buildTree(vecRightPre,vecRightIn);
	}
        return pRoot;   
}
};