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

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

注意:

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

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

返回如下的二叉树:
leetcode105. 从前序与中序遍历序列构造二叉树_第1张图片

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;   
}
};

你可能感兴趣的:(Leetcode数组)