PAT A1102 Invert a Binary Tree

题目

code

# include <cstdio>
# include <queue>
# include <stack>
# include <algorithm>
# include <cstring>
using namespace  std;
const int maxn=110;
struct node{
    int lchild,rchild;
}Node[maxn];
bool notRoot[maxn]={false};
int n,num=0;
void print(int id){
    printf("%d",id);
    num++;
    if(num<n) printf(" ");
    else printf("\n");
}
//中序遍历
void inOrder(int root){
    if(root==-1){
        return;
    }
    inOrder(Node[root].lchild);
    print(root);
    inOrder(Node[root].rchild);
}
void BFS(int root){
    queue <int> q;
    q.push(root);
    while(!q.empty()){
        int now=q.front();
        q.pop();
        print(now);
        if(Node[now].lchild!=-1) q.push(Node[now].lchild);
        if(Node[now].rchild!=-1) q.push(Node[now].rchild);
    }
}
//后序遍历
void postOrder(int root){
    if(root==-1){
        return;
    }
    postOrder(Node[root].lchild);
    postOrder(Node[root].rchild);
    swap(Node[root].lchild,Node[root].rchild);//交换左右子树
}
int stroToNum(char c){
    if(c=='-') return -1;
    else{
        notRoot[c-'0']=true;
        return c-'0';
    }
}
int findRoot(){
    for(int i=0;i<n;i++){
        if(notRoot[i]==false){
            return i;
        }
    }
    return 0;
}
int main(){
    char lchild,rchild;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%*c%c %c",&lchild,&rchild);
        Node[i].lchild=stroToNum(lchild);
        Node[i].rchild=stroToNum(rchild);
    }
    int root=findRoot();
    postOrder(root);//后序遍历 翻转二叉树
    BFS(root);
    num=0;
    inOrder(root);// 输出中序遍历的结果
    return 0;
}