[Java] 백준 - 9934번: 완전 이진 트리

9월 7일(목) - 트리 (9934번)

9934번: 완전 이진 트리

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ

 

문제점

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ

 

풀이

(C언어 풀이 활용 + List 활용 -> https://rovictory.tistory.com/121)

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

import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class Main {

    static int K;
    static int[] arr;
    static List<ArrayList<Integer>> list;
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        K = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        StringBuilder sb = new StringBuilder();

        arr = new int[(int)Math.pow(2,K) - 1];

        for(int i=0; i< arr.length; i++)
        {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        list = new ArrayList<>();
        for(int i=0; i<K; i++)
        {
            list.add(new ArrayList<>());
        }

        find_root(0, arr.length-1, 0);

        for(int i=0; i<K; i++)
        {
            for(int j : list.get(i))
            {
                sb.append(j).append(" ");
            }
            sb.append("\n");
        }

        System.out.println(sb);
    }

    static void find_root(int start, int end, int depth){

        if(depth == K)
        {
            return;
        }

        int mid = (start + end) / 2;
        list.get(depth).add(arr[mid]);

        find_root(start, mid-1, depth+1);
        find_root(mid+1, end, depth+1);
    }
}

ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ