[java 백준] 1991번 트리 순회 (++추가)

https://we1cometomeanings.tistory.com/180

[java 백준] 실버 1/ 1991번 트리 순회

위 글에서 풀이를 하나 더 보충해보려한다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
 
public class Main {
 
    static class Node {
        char ch;
        Node right;
        Node left;
 
        public Node(char ch) {
            this.ch = ch;
        }
 
    }
 
    static int N;
    static Node[] arr;
    static StringBuilder sb = new StringBuilder();
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        arr = new Node[N + 1];
        for (int i = 0; i < N; i++) {
            arr[i] = new Node((char) (i + 'A'));// arr[0]='A'로 채워짐.
        }
        for (int i = 0; i < N; i++) {
            String s = br.readLine();
            int N = s.charAt(0) - 'A';
            int L = s.charAt(2) - 'A';
            int R = s.charAt(4) - 'A';
 
            if (L != -19) {
                arr[N].left = arr[L];// N이 1일 경우 L은 3이고, R은 .(-19) 그러면 arr[1].right은 공백
            }
            if (R != -19) {
                arr[N].right = arr[R];
            }
        }
 
        preOrder(arr[0]);
        sb.append('\n');
        inOrder(arr[0]);
        sb.append('\n');
        postOrder(arr[0]);
        sb.append('\n');
        System.out.println(sb.toString());
 
    }
 
    public static void preOrder(Node node) {
        sb.append(node.ch);
        if (node.left != null) {// 예를 들어 arr[1].right은 공백이기에 이런 경우 지나갈 수 있음
            preOrder(node.left);
 
        }
        if (node.right != null) {
            preOrder(node.right);
        }
    }
 
    public static void inOrder(Node node) {
        if (node.left != null) {
            inOrder(node.left);
        }
        sb.append(node.ch);
        if (node.right != null) {
            inOrder(node.right);
        }
 
    }
 
    public static void postOrder(Node node) {
        if (node.left != null)
            postOrder(node.left);
        if (node.right != null) {
            postOrder(node.right);
 
        }
        sb.append(node.ch);
    }
}
 

 

 

그동안은 Node 생성자에 left와 right만 담았는데

이미 

for (int i = 0; i < N; i++) {

            arr[i] = new Node((char) (i + 'A'));// arr[0]='A'로 채워짐.

        }

--> 이 코드가 있기에 

 

그냥 생성자에 루트만 담아주면 된다. 

그러면 루트를 기준으로 right와 left를 호출하면 되기 때문에 편리하다! 

 

참고 사이트 

https://dos-soles.tistory.com/15

[BOJ] 1991 트리 순회 - Java