重建二叉树,主要思路是依旧输入的前序列、中序列,再分为左前序列,右中序列,右前序列,右中序列。
//#include “stdafx.h”
#include
#include
#include
#include
#include
#include <initializer_list>
#include
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}//什么意思,是否多余呢
//初始化支持 TreeNode(int x)这种方式,即 把x赋给val,left和right赋值NULL。
};
class Solution {
public:
TreeNode* reConstructBinaryTree(vector pre, vector vin) {
vector<int> pre_l, pre_r, vin_l, vin_r ;//放置递归的左右两边
//struct TreeNode *head ;//放置重建好的树,分配地址,赋值 是否和链表一样头为空呢?
//head = new TreeNode; //怎么用new???? //支持初始化
if (pre[0]==NULL) {
return 0;
}
struct TreeNode* head = new TreeNode(pre[0]);//构造函数 初值赋予的意思吗
// head = (TreeNode*)malloc(sizeof(TreeNode)); //怎么用new????
//head->val = pre[0];支持了初始化赋值
int i = 0, j = 0, k = 0,z=pre.size();
for (i = 0; i < z; i++) {
if (pre[0] == vin[i]) {
j = i;
}
}
for(k=0;k<j-1;k++){//去掉根节点重新左右排树子节
pre_l.push_back(pre[k + 1]);//第一个根节点不需要
vin_l.push_back(vin[k]);
}
for (k = j; k<z-1; k++) {
pre_r.push_back(pre[k + 1]);
vin_r.push_back(vin[k+1]);//中间根节点不需要
}
head->left = reConstructBinaryTree(pre_l, vin_l);//最开始就是一个TreeNode构造的
head->right = reConstructBinaryTree(pre_r, vin_r);
// reConstructBinaryTree(pre_l, vin_l); reConstructBinaryTree(pre_r, vin_r);
return head;
}
};
int main()
{
vector pre = { 1, 2, 4, 7, 3, 5, 6, 8 };
vector in = { 4, 7, 2, 1, 5, 3, 8, 6 };
Solution a;
struct TreeNode* head = a.reConstructBinaryTree(pre, in);
cout<<head;
}