[SWEA] 1231. [S/W 문제해결 기본] Tree - 중위순회

문제

https://www.swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV140YnqAIECFAYD&categoryId=AV140YnqAIECFAYD&categoryType=CODE&&&

SW Expert Academy

 

 

 

풀이

이 문제는 주어지는 자료를 토대로 트리를 생성해서 중위순회하여 출력하는 문제이다. 

문제 조건에서 정점의 갯수가 주어지기 때문에 정적 배열을 이용해서 트리를 구현했다. 왼쪽 자식은 인덱스*2, 오른쪽 자식은 인덱스*2 +1 를 이용해서 구현했다.

 

import java.util.*;
 
public class Solution {
    static Scanner scan = new Scanner(System.in);
    static char[] nodes;
    static int num;
    static void solution() {
        for (int i = 1; i <= 10; i++) {
            num = scan.nextInt();
            nodes = new char[num+1];
            scan.nextLine();
            for(int j=1;j<=num;j++) {
                String tmp = scan.nextLine();
                nodes[j] = (tmp.split(" ")[1]).charAt(0);
            }
            System.out.print("#"+i+" ");
            in_order(1);
            System.out.println();
        }
    }
    static void in_order(int idx) {
        if(is_valid(idx)) {
            in_order(idx*2);
            System.out.print(nodes[idx]);
            in_order(idx*2+1);
        }
    }
    static boolean is_valid(int idx) {
        if(idx>num)
            return false;
        return true;
    }
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        solution();
    }
}