LintCode 159:Construct Binary Tree from Preorder and Inorder Traversal

Description:
Given preorder and inorder traversal of a tree, construct the binary tree.


Note:

1.对前序的观察:若a在b前被遍历,则a为b的父节点或a为b的左兄弟。

2.对中序的观察:若a在b前被遍历,则a在b左边。


一开始(错误的)思路:

一开始想到树结点是不保存父节点的,因而构建树的时候也要由上之下地构建(而不能回退),所以想一个个按前序数组中的数在中序数组中比较两个结点的位置来构建树。但发现,给定两个结点在前序、中序中的关系,也无法确定这两个结点的关系,所以这种思路失败。


正确的思路:

既然无法由上之下地构建树,那就只能采用递归。(讲道理以后树的问题应该第一个想到递归这种方法了。)

递归就是要把问题拆成小问题,树的递归就是把一棵树拆成左子树和右子树。

可以发现,前序数组的第一个数一定是树的根结点,而在中序数组中,它将左、右子树分割了开来。所以通过根结点,我们重新分别构建根结点的左、右子树的前序数组和中序数组,在新分解后的数组中,每个数的前后相对关系保持不变,并且,通过观察左右子树的前序、中序遍历,我们可以发现左右子树的结点的访问是可以完全分割开的,并没有交错的部分(先访问左子树的一个节点再访问右子树的一个节点再访问左子树这样)。


Note:

1.注意在拆分数组构建新数组时,不要把root结点放进去。

2.vector<int> (begin,end)会把begin(指针)所指向的内容放进去,不会把end所指向的内容放进去。


class Solution {
		/**
		*@param preorder : A list of integers that preorder traversal of a tree
		*@param inorder : A list of integers that inorder traversal of a tree
		*@return : Root of a tree
		*/
	public:
		TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
			// write your code here
			if (preorder.size() <= 0 || inorder.size() != preorder.size())
				return nullptr;
			TreeNode* root = new TreeNode(preorder[0]);
			int i;
			for (i = 0;i<inorder.size();i++) {
				if (inorder[i] == preorder[0])
					break;
			}
			if (i >= inorder.size())
				return nullptr;
			vector<int> ino_left(inorder.begin(), inorder.begin() + i);
			vector<int> ino_right(inorder.begin() + i + 1, inorder.end());
			vector<int> pre_left;
			vector<int> pre_right;
			for (i = 1;i < ino_left.size() + 1;i++)
				pre_left.push_back(preorder[i]);
			for (i = ino_left.size() + 1;i < preorder.size();i++)
				pre_right.push_back(preorder[i]);
			root->left = buildTree(pre_left, ino_left);
			root->right = buildTree(pre_right, ino_right);
			return root;
		}
};



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值