1102 Invert a Binary Tree
题目大意:
n个节点,依次给出每个节点的左右孩子节点下标,如果没有孩子用‘-’代替,要求从右到左输出层次遍历序列,然后输出翻转二叉树后的中序遍历。
解题思路:
使用静态链表构建二叉树,标记每一个孩子节点从而找出总的根节点,然后BFS输出层次遍历。中序遍历可以提前交换节点并保存在节点结构体中,正常输出中序遍历即可。
代码如下:
#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int n,book[20];
char a,b;
struct node
{
int data;
int lchild,rchild;//翻转前
int inverl,inverr;//翻转后
}nod[20];
void bfs(int root)//层次遍历
{
int num=0;
queue<node>q;
q.push(nod[root]);
while(!q.empty())
{
node t=q.front();
cout<<t.data;
num++;
if(num<n)cout<<' ';
q.pop();
if(t.rchild!=-1)q.push(nod[t.rchild]);//注意是从右到左,所以先拓展右
if(t.lchild!=-1)q.push(nod[t.lchild]);
}
cout<<endl;
}
int num1=0;
void inorder(int root)//中序遍历
{
if(root==-1)return ;
inorder(nod[root].inverl);
cout<<nod[root].data;num1++;
if(num1<n)cout<<' ';
inorder(nod[root].inverr);
return;
}
int main()
{
// std::ios::sync_with_stdio(false);
// cin.tie(0);
// freopen("test.txt","r",stdin);
// freopen("output.txt","w",stdout);
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a>>b;
nod[i].data=i;
if(a!='-')
{
nod[i].lchild=a-'0';
book[nod[i].lchild]=1;
nod[i].inverr=a-'0';
}
else
{
nod[i].lchild=-1;
nod[i].inverr=-1;
}
if(b!='-')
{
nod[i].rchild=b-'0';
book[nod[i].rchild]=1;
nod[i].inverl=b-'0';
}
else
{
nod[i].rchild=-1;
nod[i].inverl=-1;
}
}
int root;
for(int i=0;i<n;i++)//找根节点下标
{
if(!book[i])
{
root=i;
break;
}
}
bfs(root);
inorder(root);
return 0;
}