OJ密码岛 #2341. 【模板】树的故事1

题目描述

你在 CSP-J 的赛场上睡着了。醒来发现自己在一个森林里，这个森林里有 棵树，这些树之间共有 条边相连，并且任意两棵树之间总存在一条路径能互相抵达。

更奇怪的是，这些树都可能有左孩子和右孩子。其中有一个最神圣的树，它的编号为 ，它不是任何一个其他树的孩子。

现在树神告诉了你三种遍历这些树的方法：

先序遍历：先标记当前这棵树，然后，如果它有左孩子，遍历它的左孩子，接着，如果它有右孩子，遍历它的右孩子。
中序遍历：如果当前的树有左孩子，先遍历它的左孩子，然后，标记当前这棵树，接着，如果它有右孩子，遍历它的右孩子。
后序遍历：如果当前的树有左孩子，先遍历它的左孩子，然后，如果它有右孩子，遍历它的右孩子，接着，标记当前这棵树。
树神会指定一种遍历方法，让你输出树被标记的顺序。

输入格式

输入共 n+1 行。

第一行两个正整数 n，t分别表示树的个数和遍历的序号。

接下来 n 行，第 i+1 行两个正整数 l[i]，r[i]表示第 个树的左孩子和右孩子。如果没有这个孩子，这个数是 0。

输出格式

输出共一行，为这棵树的相应遍历序列。

样例

输入样例
5 2
2 3
0 4
5 0
0 0
0 0

输出样例
2 4 1 5 3

代码

#include <iostream>
using namespace std;
int a[10005][2],t;
//父节点的编号
//a[i][0] i节点的左孩子
//a[i][0] i节点的右孩子
void dfs1(int x)
{
	if(x==0) return;
	cout << x << ' ';
	dfs1(a[x][0]);
	dfs1(a[x][1]);
}
void dfs2(int x)
{
	if(x==0) return;
	dfs2(a[x][0]);
	cout << x << ' ';
	dfs2(a[x][1]);
}
void dfs3(int x)
{
	if(x==0) return;
	dfs3(a[x][0]);
	dfs3(a[x][1]);
	cout << x << ' ';
}
int main()
{
	int n;
	cin >> n >> t;
	for(int i=1;i<=n;i++) cin >> a[i][0] >> a[i][1];
	if(t==1) dfs1(1);
	if(t==2) dfs2(1);
	if(t==3) dfs3(1);
	return 0;
}