PTA 1102 Invert a Binary Tree

题目描述

分析：

交换每个子树的左右孩子，完后输出。题目给出的是每个结点的左右孩子，首先找到根结点，也就是不是任何结点的孩子的那个就是根结点。完后后序遍历的同时交换左右子树，最后输出层序遍历和中序遍历。

#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;

struct node{
	int lchild,rchild;
}no[105];

bool root[105]={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(no[root].lchild);
	print(root);
	inOrder(no[root].rchild);
}

void BFS(int root){
	queue<int> q;
	q.push(root);
	while(!q.empty()){
		int now=q.front();
		q.pop();
		print(now);
		if(no[now].lchild!=-1) q.push(no[now].lchild);
		if(no[now].rchild!=-1) q.push(no[now].rchild);
	}
}
void postOrder(int root){
	if(root==-1)
	return ;
	postOrder(no[root].lchild);
	postOrder(no[root].rchild);
	swap(no[root].lchild,no[root].rchild);
}

int strTonum(char c){
	if(c=='-')
	return -1;
	else{
		root[c-'0']=true;
		return c-'0';
	}
}
int findRoot(){
	for(int i=0;i<n;i++){
		if(!root[i])
		return i;
	}
}
int main(){
	char lchild,rchild;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%*c%c %c",&lchild,&rchild);
		no[i].lchild=strTonum(lchild);
		no[i].rchild=strTonum(rchild);
	}
	int root=findRoot();
	postOrder(root);
	BFS(root);
	num=0;
	inOrder(root);
	return 0;
}