PAT.A1102 Invert a Binary Tree

返回目录

题意

给出一颗编号节点为0-n-1的二叉树，给出每个节点的孩子结点编号，输出把该二叉树反转后的层次遍历和中序遍历序列。

样例（可复制）

8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6

//output
3 7 2 6 4 0 5 1
6 5 7 4 3 2 0 1

注意点

1. 这里对样例的输入进行解释，第一行输入的是n个结点，之后每行依次输入0-n-1号结点的左右孩子编号，由于本题结点使用编号给出，固使用静态树的struct进行存储较方便
2. 本题输入时使用scanf("%*c%c %c",&a,&b);，其中%*c会吸收上一行输入的换行符“\n"
3. 没有出现在所有结点的左右孩子编号的编号为根结点编号，固建立一个flag[]对输入中出现的的编号做标记，之后遍历一遍flag即可得到根节点编号

#include<bits/stdc++.h>
using namespace std;

struct Node{
	int lc,rc;
}node[15]; 
bool flag[15]={false};
int n,num=0;
void reverse(int root){
	if(root==-1)return;
	reverse(node[root].lc);
	reverse(node[root].rc);
	swap(node[root].lc,node[root].rc);
}
void BFS(int root){
	queue<int> q;
	q.push(root);
	while(!q.empty()){
		int now=q.front();
		q.pop();
		if(node[now].lc!=-1)q.push(node[now].lc);
		if(node[now].rc!=-1)q.push(node[now].rc);
		printf("%d",now);
		num++;
		if(num<n)printf(" ");
	}
	printf("\n");
}
void inorder(int root){
	if(root==-1) return;
	if(node[root].lc!=-1)inorder(node[root].lc);
	printf("%d",root);
	num++;
	if(num<n)printf(" ");
	if(node[root].rc!=-1)inorder(node[root].rc);
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		char a,b;
		scanf("%*c%c %c",&a,&b);
		if(a=='-')node[i].lc=-1;
		else{
			node[i].lc=a-'0';
			flag[a-'0']=true;
		} 
		if(b=='-')node[i].rc=-1;
		else{
			node[i].rc=b-'0';
			flag[b-'0']=true;
		}
	}
	int root;
	for(int i=0;i<n;i++){//得到根节点编号
		if(!flag[i])root=i;
	}
	reverse(root);//反转 
	BFS(root);//输出层次遍历
	num=0;
	inorder(root);//输出中序遍历 
    return 0;
}