PAT(甲级)A-1102 Invert a Binary Tree(预处理反转)

PAT A-1102 Invert a Binary Tree(预处理反转)

在读入节点左右孩子信息时就完成反转，在之后 层序遍历 和 中序遍历 时就不需要再做特殊处理

#include<iostream>
#include<string>
#include<queue>
using namespace std;

const int maxn=15;
int n;
struct node{
    int data;
    int lchild=-1,rchild=-1;//空节点初始化为-1
}A[maxn];

bool ischild[maxn]={false};//记录节点是否为孩子节点
int findroot(){
    for(int i=0;i<n;i++)//遍历找出从未成为孩子的节点,即为根节点
        if(ischild[i]==false)
            return i;
}

void BFS(int root){//层序遍历并输出
    int numq=0;
    queue<int> q;
    q.push(root);
    while(!q.empty()){
        int now=q.front();
        q.pop();
        cout<<A[now].data;
        if(++numq!=n)
            cout<<' ';
        if(A[now].lchild!=-1)
            q.push(A[now].lchild);
        if(A[now].rchild!=-1)
            q.push(A[now].rchild);
    }
    cout<<endl;
}

int numin;//已输出的节点数
void inorder(int root){//中序遍历并输出
    if(A[root].lchild!=-1)
        inorder(A[root].lchild);

    cout<<A[root].data;
    if(++numin!=n)
        cout<<' ';
    
    if(A[root].rchild!=-1)
        inorder(A[root].rchild);
}

int main()
{
    cin>>n;
    getchar();
    for(int i=0;i<n;i++){
        A[i].data=i;
        string s;
        getline(cin,s);
        
        if(s[0]!='-'){
            A[i].rchild=s[0]-'0';//将左孩子存入右孩子,即完成反转
            ischild[A[i].rchild]=true;//已成为孩子节点,标记为true
        }
        if(s[2]!='-'){
            A[i].lchild=s[2]-'0';//将右孩子存入左孩子,即完成反转
            ischild[A[i].lchild]=true;//已成为孩子节点,标记为true
        }
    }
    int root=findroot();
    BFS(root);
    inorder(root);
    cout<<endl;
    
    return 0;
}