1521:二叉樹的映象 @jobdu
阿新 • • 發佈:2019-02-10
題目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_; } } }