重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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:
    void dfs(TreeNode * node,vector<int>& pre, int a1,int a2,vector<int>& vin, int b1, int b2) {
	if (a1 > a2 || b1 > b2) {//当左大于右，即没有节点，返回
		return;
	}
	int f1 = pre[a1];//记录当前一段前序的根值
	int i = 0;
	for ( i = b1; i <= b2; i++) {//寻找在中序遍历中根值的位置
		if (f1 == vin[i]) {//找到根值，跳出
			break;
		}
	}
	int l_a = i-b1;//左节点个数，即i下标距离中序遍历开始点的距离
	int r_a = b2 - i;//右节点个数，同上不过是左右的差别，因为i下标代表的是根节点位置
	if (l_a > 0) {//有左节点则加，并继续递归
		TreeNode* temp = (TreeNode*)malloc(sizeof(TreeNode));
		temp->val = pre[a1 + 1];
		temp->left = temp->right = NULL;
		node->left = temp;
		dfs(temp,pre, a1 + 1, a1 + l_a, vin, b1, i - 1);
	}
	if (r_a > 0) {//有右节点则加，并继续递归
		TreeNode* temp = (TreeNode*)malloc(sizeof(TreeNode));
		temp->val = pre[a1 + l_a + 1];
		temp->left = temp->right = NULL;
		node->right = temp;
		dfs(temp,pre, a1 + l_a + 1, a2, vin, i + 1, b2);
	}
	}
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin) {
	TreeNode* head = NULL;
	if (pre.size() == 0) {
		return head;
	}
	TreeNode* temp = (TreeNode*)malloc(sizeof(TreeNode));
	temp->val = pre[0];
	temp->left = temp->right = NULL;
	head = temp;
	dfs(head,pre, 0, pre.size() - 1,vin, 0, vin.size() - 1);
	return head;
}
          
};

本身这道题不难，在应用递归思想解决之前，我们只要处理好两件事

1. 由于在原数组上进行操作，我们要计算好下标的位置，即每次递归时，有效数据长度

2.想好实现数值记录，即二叉树建立位置或者说方式