北邮08复试模拟problem C

 

第三题：二叉树前序遍历 Submit: 2541    Accepted:674 Time Limit: 1000MS  Memory Limit: 65535K

Description
给定一棵有n个结点的二叉树，结点的编号为0~n-1。请你编写程序输出二叉树的前序遍历序列。

Input
输入的第一行是一个正整数t(1 < t < 20)，表示有t组测试用例。
对于每组测试用例，第一行是一个整数n(0 < n < 20)，表示二叉树结点个数。第二行是一个数r(0≤r≤n-1)，二叉树根结点的编号。
后面有n-1行，表示二叉树n-1条边的信息。每行三个数a,b,c，三个数间由空格隔开，其中0≤a,b≤n-1且a≠b, c为0或1。a表示边的起点，b表示边的终点。如果c为0，表示b是a的左儿子；如果c为1，表示b是a的右儿子。

Output
对于每组测试用例输出一行，即：该二叉树的前序遍历序列，两个节点编号之间留一个空格。

Sample Input

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

Sample Output

2 0 1
0 1 3 4 2 5 6

Hint
由于是计算机自动判题，请严格按照题目的描述输入输出，不要有任何多余的字符出现，尤其是输出行的行首和行尾都不要有多余的空格

Source
import java.util.*; class treeNode{ private treeNode left; public treeNode getLeft() { return left; } public void setLeft(treeNode left) { this.left = left; } public treeNode getRight() { return right; } public void setRight(treeNode right) { this.right = right; } public int getValue() { return value; } public void setValue(int value) { this.value = value; } private treeNode right; private int value; public treeNode(int value){ this.value=value; } } public class tree { static ArrayList<Object> rl=new ArrayList<Object> (); public static void main(String[] args) { Scanner cin = new Scanner(System.in); int t=cin.nextInt(); for(int i=0;i<t;i++){ int n=cin.nextInt(); int r=cin.nextInt(); ArrayList<treeNode> tn=new ArrayList<treeNode>(); treeNode root=new treeNode(r); tn.add(0, root); int count=1; for(int k=0;k<n-1;k++){ int[] temp=new int[3]; for(int ii=0;ii<3;ii++){ temp[ii]=cin.nextInt(); } treeNode node=new treeNode(temp[1]); tn.add(count, node); int index=getIndex(tn,temp[0]); if(temp[2]==0){ tn.get(index).setLeft(tn.get(count)); } else{ tn.get(index).setRight(tn.get(count)); } count++; } preorder(tn.get(0)); rl.add("/n"); } for(int k=0;k<rl.size()-1;k++){ System.out.print(rl.get(k)); if(rl.get(k+1).equals("/n")||rl.get(k).equals("/n")){ continue; } System.out.print(" "); } System.out.println(""); } private static int getIndex(ArrayList<treeNode> tn, int i) { for(int t=0;t<tn.size();t++){ if(tn.get(t).getValue()==i){ return t; } } return 0; } private static void preorder(treeNode p) { if (p != null) { visit(p); preorder(p.getLeft()); preorder(p.getRight()); } } private static void visit(treeNode p) { rl.add(p.getValue()); } }