PAT 1102 Invert a Binary Tree（找根+翻转树+层序）

传送门

给一颗树树要求输出反转二叉树的层序遍历和中序遍历。

先找出根节点，根节点即不是其他节点孩子的节点。

建树，再翻转（先递归右子树）参考柳神的index标记法可以不用bfs就能层序遍历

#include <iostream>
#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<n;i++)
#define sca(x) scanf("%d",&x)
#define sca2(x,y) scanf("%d%d",&x,&y)
#define scl(x) scanf("%lld",&x)
#define pri(x) printf("%d\n",x)
#define pri2(x,y) printf("%d %d\n",x,y)
#define prs(x) printf("%s\n",(x))
#define prl(x) printf("%lld\n",x)
#define ll long long
#define PII pair<int,int>
#define eps 1e-6
#define inf 1e17
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=1e6+5;
struct node{
    int id,l,r,index,level;
}tree[maxn];
int n;
string l,r;
int flag[maxn];
vector<node>v1,v2;
void dfs(int rt,int level,int index){
    if(tree[rt].r!=-1)
        dfs(tree[rt].r,level+1,index*2+2);
    tree[rt].level=level,tree[rt].index=index;
    v1.push_back(tree[rt]);
    if(tree[rt].l!=-1)
        dfs(tree[rt].l,level+1,index*2+1);
}
bool cmp(node a,node b){
    if(a.level==b.level) return a.index>b.index;
    return a.level<b.level;
}
int main(){
    sca(n);
    rep(i,0,n){
        cin>>l>>r;
        if(l[0]=='-'){
            tree[i].l=-1;
        }
        else{
            tree[i].l=l[0]-'0';
            flag[tree[i].l]=1;
        }
        if(r[0]=='-'){
            tree[i].r=-1;
        }
        else{
            tree[i].r=r[0]-'0';
            flag[tree[i].r]=1;
        }
        tree[i].id=i;
    }

    int root=0;
    rep(i,0,n){
        if(!flag[i]){
            root=i;break;
        }
    }
    dfs(root,0,0);
    v2=v1;
    sort(v2.begin(),v2.end(),cmp);
    for(int i=0;i<v2.size();i++){
        printf("%d",v2[i].id);
        if(i==v2.size()-1) puts("");
        else printf(" ");
    }
    for(int i=0;i<v1.size();i++){
        printf("%d",v1[i].id);
        if(i==v1.size()-1) puts("");
        else printf(" ");
    }
    return 0;
}