玩转二叉树
程序员文章站
2022-05-17 07:54:36
...
先序和中序遍历序列
具体涉及问题有建二叉树、镜面反转、bfs(层次遍历).
#include<bits/stdc++.h>
using namespace std;
struct node
{
int l,r;
}nodes[32];
int pre[32],in[32];
int n;
int build(int a,int b,int c,int d)
{
int root=pre[a];
int m=c;
while(in[m]!=root)
m++;
if(m==c)
nodes[root].l=-1;
else
nodes[root].l=build(a+1,a+m-c,c,m-1);
if(m==d)
nodes[root].r=-1;
else
nodes[root].r=build(a+m-c+1,b,m+1,d);
return root;
}
void bfs(int root)
{
if(root==-1)
return;
queue<int> q;
q.push(root);
while(!q.empty())
{
int m=q.front();
if(m==root)
cout<<m;
else
cout<<" "<<m;
if(nodes[m].l!=-1)
q.push(nodes[m].l);
if(nodes[m].r!=-1)
q.push(nodes[m].r);
q.pop();
}
}
void Swap(int root)
{
int tmp=nodes[root].l;
nodes[root].l=nodes[root].r;
nodes[root].r=tmp;
}
void revswap(int root)
{
if(root==-1)
return;
revswap(nodes[root].l);
revswap(nodes[root].r);
Swap(root);
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
cin>>in[i];
for(int i=0;i<n;i++)
cin>>pre[i];
int root=build(0,n-1,0,n-1);
revswap(root);
bfs(root);
return 0;
}
上一篇: 玩转二叉树
下一篇: Vue计算属性和侦听器