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;
}