PAT——1102 Invert a Binary Tree 甲级

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805365537882112

题意

invert 反转，即将树中每个节点的左右孩子互换。在已知每个孩子的左右节点后，输出该树的层序遍历和中序遍历

代码解析

根节点的寻找方式为没有当过孩子的节点就是根节点

层序遍历通过bfs，套用模板得出答案

中序遍历通过dfs，按照左中右的顺序存入ans2中即可

注意，结构体中，左右孩子的默认值要设为-1而不能是0，因为0是节点之一并不能作为空孩子的判断值。

AC代码

#include<bits/stdc++.h>
using namespace std;
struct Node{
	int l=-1,r=-1;
}node[12];
int n,isSon[12],Root=0;
vector<int> ans1,ans2;
void bfs(int root)
{
	queue<int> q;
	q.push(root);
	while(q.size())
	{
		int t=q.front();
		q.pop();
		ans1.push_back(t);
		if(node[t].l!=-1) q.push(node[t].l);
		if(node[t].r!=-1) q.push(node[t].r);
	}
}
void dfs(int root)
{
	if(node[root].l!=-1) dfs(node[root].l);
	ans2.push_back(root);
	if(node[root].r!=-1) dfs(node[root].r);
}
int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
	{
		string a,b;
		cin>>a>>b;
		if(a!="-") 
		{
			node[i].r=stoi(a);
			isSon[stoi(a)]=1;
		}
		if(b!="-") 
		{
			node[i].l=stoi(b);
			isSon[stoi(b)]=1;
		}
	}
	while(isSon[Root]) Root++;
	bfs(Root);
	dfs(Root);
	for(int i=0;i<ans1.size();i++)
	{
		if(i) cout<<" ";
		cout<<ans1[i];	
	}	
	cout<<endl;
	for(int i=0;i<ans2.size();i++)
	{
		if(i) cout<<" ";
		cout<<ans2[i];	
	}	
}