백준 1991 트리 순회 Java

: ) YOUNG·2023년 12월 27일
1

알고리즘

목록 보기
286/411
post-thumbnail

백준 1991번
https://www.acmicpc.net/problem/1991

문제



생각하기


  • 트리 자료구조 문제이다.

  • 트리 자료구조를 만들고, 순회에 따라 출력만 하면되는데, 사실상 입력 받는게 좀 어렵고, 출력은 정해져있는 방식을 외우기만 하면 되서 어렵지 않다.


동작



결과


코드


배열 구조 트리


import java.io.*;
import java.util.StringTokenizer;

public class Main {

    // input
    private static BufferedReader br;

    // variables
    private static int N;
    private static Node[] tree;
    private static StringBuilder sb;

    private static class Node {
        char cur;
        Node left;
        Node right;

        private Node(char cur) {
            this.cur = cur;
            this.left = null;
            this.right = null;
        }
    } // End of Node class

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

        input();

        bw.write(solve());
        bw.close();
    } // End of main()

    private static String solve() {

        // 전위 순회
        preOrder(tree[0]);
        sb.append('\n');

        // 중위 순회
        inOrder(tree[0]);
        sb.append('\n');

        // 후위 순회
        postOrder(tree[0]);
        sb.append('\n');
        
        return sb.toString();
    } // End of solve()

    private static void preOrder(Node node) {
        if (node == null) return;
        sb.append(node.cur);
        preOrder(node.left);
        preOrder(node.right);
    } // End of preOrder()

    private static void inOrder(Node node) {
        if (node == null) return;
        inOrder(node.left);
        sb.append(node.cur);
        inOrder(node.right);
    } // End of inOrder()

    private static void postOrder(Node node) {
        if (node == null) return;
        postOrder(node.left);
        postOrder(node.right);
        sb.append(node.cur);
    } // End of postOrder()

    private static void input() throws IOException {
        N = Integer.parseInt(br.readLine());
        tree = new Node[N + 1];
        sb = new StringBuilder();


        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            char cur = st.nextToken().charAt(0);
            char left = st.nextToken().charAt(0);
            char right = st.nextToken().charAt(0);

            if (tree[cur - 'A'] == null) {
                tree[cur - 'A'] = new Node(cur);
            }

            if (left != '.') {
                tree[left - 'A'] = new Node(left);
                tree[cur - 'A'].left = tree[left - 'A'];
            }

            if (right != '.') {
                tree[right - 'A'] = new Node(right);
                tree[cur - 'A'].right = tree[right - 'A'];
            }
        }
    } // End of input()
} // End of Main class


객체 구조 트리


import java.io.*;
import java.util.StringTokenizer;

public class Main {

    // input
    private static BufferedReader br;

    // variables
    private static int N;
    private static Node tree;
    private static StringBuilder sb;

    private static class Node {
        char cur;
        Node left;
        Node right;

        private Node(char cur) {
            this.cur = cur;
            this.left = null;
            this.right = null;
        }
    } // End of Node class

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

        input();

        bw.write(solve());
        bw.close();
    } // End of main()

    private static String solve() {
        preOrder(tree);
        sb.append('\n');

        inOrder(tree);
        sb.append('\n');

        postOrder(tree);
        sb.append('\n');

        return sb.toString();
    } // End of solve()

    private static void preOrder(Node node) {
        if (node == null) return;
        sb.append(node.cur);
        preOrder(node.left);
        preOrder(node.right);
    } // End of preOrder()

    private static void inOrder(Node node) {
        if (node == null) return;
        inOrder(node.left);
        sb.append(node.cur);
        inOrder(node.right);
    } // End of inOrder()

    private static void postOrder(Node node) {
        if (node == null) return;
        postOrder(node.left);
        postOrder(node.right);
        sb.append(node.cur);
    } // End of postOrder()

    private static void insert(Node parent, char cur, char left, char right) {
        if (parent.cur == cur) {
            if (left != '.') {
                parent.left = new Node(left);
            }

            if (right != '.') {
                parent.right = new Node(right);
            }
        } else {
            if (parent.left != null) {
                insert(parent.left, cur, left, right);
            }

            if (parent.right != null) {
                insert(parent.right, cur, left, right);
            }
        }
    } // End of insert()

    private static void input() throws IOException {
        N = Integer.parseInt(br.readLine());
        tree = new Node('A');
        sb = new StringBuilder();

        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            char cur = st.nextToken().charAt(0);
            char left = st.nextToken().charAt(0);
            char right = st.nextToken().charAt(0);
            insert(tree, cur, left, right);
        }
    } // End of input()
} // End of Main class


0개의 댓글