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

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

注意:

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

例如，给出
中序遍历 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;    	
}
};