1. 程式人生 > >1521:二叉樹的映象 @jobdu

1521:二叉樹的映象 @jobdu

題目1521:二叉樹的映象

時間限制:1 秒

記憶體限制:128 兆

特殊判題:

提交:661

解決:165

題目描述:

輸入一個二叉樹,輸出其映象。

輸入:

輸入可能包含多個測試樣例,輸入以EOF結束。
對於每個測試案例,輸入的第一行為一個整數n(0<=n<=1000,n代表將要輸入的二叉樹節點的個數(節點從1開始編號)。接下來一行有n個數字,代表第i個二叉樹節點的元素的值。接下來有n行,每行有一個字母Ci。
Ci=’d’表示第i個節點有兩子孩子,緊接著是左孩子編號和右孩子編號。
Ci=’l’表示第i個節點有一個左孩子,緊接著是左孩子的編號。
Ci=’r’表示第i個節點有一個右孩子,緊接著是右孩子的編號。
Ci=’z’表示第i個節點沒有子孩子。

輸出:

對應每個測試案例,
按照前序輸出其孩子節點的元素值。
若為空輸出NULL。

樣例輸入:
7
8 6 10 5 7 9 11
d 2 3
d 4 5
d 6 7
z
z
z
z
樣例輸出:
8 10 11 9 6 7 5

特別注意的是

if(tree[i] == null){		// 特別注意這裡!當且僅當為空時才需要賦值!
					tree[i] = new TreeNode(tmp[i]);
				}
因為之後的程式碼預先改變了這裡的值,所以如果再賦值就會出現指標問題!花了近半個小時debug這個問題!
還有一個就是最後一個元素後面不能列印空格!由於用遞迴列印,所以這裡用一個全域性變數記錄列印了多少次,對最後一次特殊處理。
import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Arrays;
import java.util.Scanner;

public class S19 {
	static int loop = 0;
	public static void main(String[] args) throws FileNotFoundException {
		BufferedInputStream in = new BufferedInputStream(new FileInputStream("in.in"));
		System.setIn(in);
		Scanner cin = new Scanner(System.in);
		
		while (cin.hasNextInt()) {
			int n = cin.nextInt();
			loop = n;
			int[] tmp = new int[n];
			for(int i=0; i<n; i++){
				tmp[i] = cin.nextInt();
			}
//			System.out.println(Arrays.toString(tmp));
			if(n == 0){
				System.out.println("NULL");
				continue;
			}
			cin.nextLine();
			TreeNode[] tree = new TreeNode[n];
			for(int i=0; i<n; i++){
				String s = cin.nextLine();
//System.out.println(s);
				String[] ss = s.split(" ");
				if(tree[i] == null){		// 特別注意這裡!當且僅當為空時才需要賦值!
					tree[i] = new TreeNode(tmp[i]);
				}
				if(ss[0].equals("d")){
					int left = Integer.parseInt(ss[1]);
					int right = Integer.parseInt(ss[2]);
					if(tree[left-1]==null)
						tree[left-1] = new TreeNode(tmp[left-1]);
					if(tree[right-1]==null)
						tree[right-1] = new TreeNode(tmp[right-1]);
					tree[i].left = tree[left-1];
					tree[i].right = tree[right-1];
				}else if(ss[0].equals("l")){
					int left = Integer.parseInt(ss[1]);
					tree[left-1] = new TreeNode(tmp[left-1]);
					tree[i].left = tree[left-1];
				}else if(ss[0].equals("r")){
					int right = Integer.parseInt(ss[1]);
					tree[right-1] = new TreeNode(tmp[right-1]);
					tree[i].right = tree[right-1];
				}else if(ss[0].equals("z")){
					
				}
			}
			
//			preorder(tree[0]);
			TreeNode mirrored = mirror(tree[0]);
			if(mirrored != null){
				preorder(mirrored);
				System.out.println();
			}
		}
	}
	
	private static void preorder(TreeNode root) {
		if(root != null){
			loop--;
			if(loop == 0){
				System.out.print(root.val);
			}else{
				System.out.print(root.val + " ");
			}
			preorder(root.left);
			preorder(root.right);
		}
	}

	private static TreeNode mirror(TreeNode root) {
		if(root == null){
			return null;
		}
		
		TreeNode tmp = root.left;
		root.left = mirror(root.right);
		root.right = mirror(tmp);
		return root;
	}

	public static class TreeNode{
		int val;
		TreeNode left, right;
		public TreeNode(int val_){
			val = val_;
		}
	}
}