主要借助这道题标记一下如何完整的数组建树,翻之前的博客发现只写了递归版本,找后序但是没有完整遍历
核心构树代码:
主要思想和之前相同,找到前序遍历在中序遍历中的位置,之后递归重构左,重构右
注意这里先后的重构顺序与题目给出的序列有关
这两年的天梯赛都参加了,似乎没有这类纯裸树的题了,可能姥姥觉得大家都已经研究明白了吧
struct node
{
int l;
int r;
};
struct node tree[1005];
int pre[1005];
int in[1005];
int post[1005];
int n;
int pos;
int rec(int l,int r)
{
if(l>=r)
{
return -1;
}
int root = post[pos--];
int m;
for(int i=0;i<n;i++)
{
if(in[i]==root)
{
m = i;
break;
}
}
tree[root].r = rec(m+1,r);
tree[root].l = rec(l,m);
return root;
}
#include <iostream>
#include <queue>
using namespace std;
struct node
{
int l;
int r;
};
struct node tree[1005];
int pre[1005];
int in[1005];
int post[1005];
int n;
int pos;
int rec(int l,int r)
{
if(l>=r)
{
return -1;
}
int root = post[pos--];
int m;
for(int i=0;i<n;i++)
{
if(in[i]==root)
{
m = i;
break;
}
}
tree[root].r = rec(m+1,r);
tree[root].l = rec(l,m);
return root;
}
queue <int> que;
int start;
int space = 0;
void levelorder(int root)
{
while(que.size())
{
int a = que.front();
que.pop();
if(space!=0)
{
printf(" ");
}
space ++;
printf("%d",a);
if(tree[a].l!=-1)
que.push(tree[a].l);
if(tree[a].r!=-1)
que.push(tree[a].r);
}
}
int main(void)
{
scanf("%d",&n);
for(int i=0;i<n;i++)//给出后序和中序
{
scanf("%d",&post[i]);
}
for(int i=0;i<n;i++)
{
scanf("%d",&in[i]);
}
pos = n-1;
start = rec(0,n);
que.push(start);
levelorder(start);
}