leetcode106. 从中序与后序遍历序列构造二叉树

最新推荐文章于 2024-07-18 10:27:47 发布

一条特立独行的狗、

最新推荐文章于 2024-07-18 10:27:47 发布

阅读量106

点赞数

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

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

版权

Leetcode数组专栏收录该内容

68 篇文章 0 订阅

订阅专栏

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

注意:

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

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

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


class Solution {
public:    
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if( postorder.empty() || inorder.empty() )            
        	return NULL;        
        int nRootVal = postorder[postorder.size()-1];        
        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> vecLeftPost(postorder.begin(),postorder.begin()+nIdx);            
		vector<int> vecLeftIn(inorder.begin(),inorder.begin()+nIdx);
		vector<int> vecRightPost(postorder.begin()+nIdx,postorder.end()-1);
		vector<int> vecRightIn(inorder.begin()+nIdx+1,inorder.end());
		pRoot->left = buildTree(vecLeftIn,vecLeftPost);            												
		pRoot->right = buildTree(vecRightIn,vecRightPost);
        }
        return pRoot;    	
}
};