[JAVA-자바] 1991번: 트리 순회

 

소스코드:

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;

public class Main {
    static BufferedReader br;
    static BufferedWriter bw;

    static int N;
    static char[][] node;

    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        bw = new BufferedWriter(new OutputStreamWriter(System.out));

        N = Integer.parseInt(br.readLine());
        node = new char[N][3];
        for (int i = 0; i < N; i++) {
            String input = br.readLine();
            node[i][0] = input.charAt(0);
            node[i][1] = input.charAt(2);
            node[i][2] = input.charAt(4);
        }
        br.close();

        preorder(node[0][0]);
        bw.newLine();
        inorder(node[0][0]);
        bw.newLine();
        postorder(node[0][0]);

        bw.close();
    }

    private static char[] findNode(char name) {
        char[] find = new char[3];
        for (int i = 0; i < N; i++) {
            if (node[i][0] == name) {
                find = node[i];
                break;
            }
        }
        return find;
    }

    private static void preorder(char root) throws IOException {
        char[] currentNode = findNode(root);

        bw.write(currentNode[0]);

        if (currentNode[1] != '.') {
            preorder(currentNode[1]);
        }
        if (currentNode[2] != '.') {
            preorder(currentNode[2]);
        }
    }

    private static void inorder(char root) throws IOException {
        char[] currentNode = findNode(root);

        if (currentNode[1] != '.') {
            inorder(currentNode[1]);
        }

        bw.write(currentNode[0]);

        if (currentNode[2] != '.') {
            inorder(currentNode[2]);
        }
    }

    private static void postorder(char root) throws IOException {
        char[] currentNode = findNode(root);

        if (currentNode[1] != '.') {
            postorder(currentNode[1]);
        }

        if (currentNode[2] != '.') {
            postorder(currentNode[2]);
        }

        bw.write(currentNode[0]);
    }
}

 

글의 내용 중 잘못된 점이나 수정이 필요한 부분, 혹은 궁금한 사항이 있다면 언제든 댓글로 남겨주시면 감사하겠습니다.

여러분의 피드백은 더 나은 글을 작성하는 데 큰 도움이 됩니다. 감사합니다.