欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

玩转二叉树

程序员文章站 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;
}

相关标签: 平时训练