《劍指offer》重建二叉樹

【 聲明：版權所有，轉載請标明出處，請勿用于商業用途。  聯系信箱：[email protected]】

題目連結：http://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6?rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

題目描述

輸入某二叉樹的前序周遊和中序周遊的結果，請重建出該二叉樹。假設輸入的前序周遊和中序周遊的結果中都不含重複的數字。例如輸入前序周遊序列{1,2,4,7,3,5,6,8}和中序周遊序列{4,7,2,1,5,3,8,6}，則重建二叉樹并傳回。

思路

考察基礎的二叉樹周遊，使用遞歸實作

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution
{
	public:
		struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in)
		{
			int lenp = pre.size();
			int leni = in.size();
			if(lenp==0 || leni==0 || lenp!=leni)
				return nullptr;*

			TreeNode *root = new TreeNode(pre[0]);
			int mid = find(in.begin(),in.end(),pre[0])-in.begin();

			vector<int> left_pre(pre.begin()+1,pre.begin()+1+mid);
			vector<int> left_in(in.begin(),in.begin()+mid);
			vector<int> right_pre(pre.begin()+1+mid,pre.end());
			vector<int> right_in(in.begin()+mid+1,in.end());

			root->left = reConstructBinaryTree(left_pre,left_in);
			root->right = reConstructBinaryTree(right_pre,right_in);

			return root;
		}
};