【二叉树】04 给出前序遍历和中序遍历复原二叉树

///给出前序遍历和中序遍历复原二叉树

TreeNode* rebuild(int* pdata, int* idata, int n) {
//假设给出的前序中序都是数组,pdata 前序；idata中序；n是长度
	if (n <= 0) {
		return NULL;
	}
	TreeNode* root = new TreeNode(*pdata);

	int* iter;
	for (iter = idata; iter < idata + n; iter++) {
		if (*iter == *pdata) {
		//找前序中的节点在中序的位置，然后把中序分成左右两个
			break;
		}
	}

	int k = iter - idata;//找出把中序排序分成两部分左边部分的长度
	root->left = rebuild(pdata + 1, idata, k);
	root->right = rebuild(pdata + k + 1, idata + 1, n - k - 1);

	return root;
}