PAT甲题题解-1102. Invert a Binary Tree (25)-（建树，水题）

　　就是把输入给的左孩子右孩子互换一下，然后输出层次遍历和中序遍历。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cstdio>
#include <queue>

using namespace std;
const int maxn=15;
bool first=true;
struct Node{
    int id;
    int left;
    int right;
}node[maxn];
int vis[maxn]; //没出现的即是根节点
//层次遍历
void level_order(int i){
    queue<Node>q;
    q.push(node[i]);
    Node tmp;
    while(!q.empty()){
        tmp=q.front();
        q.pop();
        if(first){
            first=false;
            printf("%d",tmp.id);
        }
        else
            printf(" %d",tmp.id);
        int l=tmp.left;
        int r=tmp.right;
        if(l!=-1)
            q.push(node[l]);
        if(r!=-1)
            q.push(node[r]);
    }
}
//中序遍历
void in_order(int i){
    if(i==-1)
        return;
    int l=node[i].left;
    int r=node[i].right;
    in_order(l);
    if(first){
        printf("%d",i);
        first=false;
    }
    else{
        printf(" %d",i);
    }
    in_order(r);
}
int main()
{
    int n;
    char str1[10],str2[10];
    memset(vis,0,sizeof(vis));
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%s %s",str1,str2);
        node[i].id=i;
        if(str1[0]=='-')
            node[i].right=-1;
        else{
            node[i].right=str1[0]-'0';
            vis[str1[0]-'0']=1;
        }
        if(str2[0]=='-')
            node[i].left=-1;
        else{
            node[i].left=str2[0]-'0';
            vis[str2[0]-'0']=1;
        }
    }
    int root;
    for(int i=0;i<n;i++){
        if(!vis[i]){
            root=i;
            break;
        }
    }
    first=true;
    level_order(root);
    printf("\n");
    first=true;
    in_order(root);
    return 0;
}