BOJ9934: 완전 이진 트리

https://www.acmicpc.net/problem/9934

  • in-order를 이해하는 문제. 반반씩 쪼개서 풀면 된다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#include <bits/stdc++.h>
using namespace std;


class Problem {
  int N;
  vector<int> nodes;
  vector<vector<int>> depthToNode;
public:
  Problem(int n): N(n), nodes((1<<n)-1), depthToNode(n) {}
  void input() {
    for(int i = 0; i < (1<<N)-1; i++)
      cin >> nodes[i];
  }
  void dfs(int p, int depth) {
    if(depth == N) return;
    int size = (1 << N-depth) - 1;
    int mid = size / 2 + p;
    dfs(p, depth+1);
    depthToNode[depth].push_back(nodes[mid]);
    dfs(mid+1, depth+1);
  }
  void print() {
    for(auto &v: depthToNode) {
      for(int x: v)
        printf("%d ", x);
      printf("\n");
    }
  }
};

int main() {
  int N;
  ios::sync_with_stdio(false);
  cin >> N;
  Problem pr(N);
  pr.input();
  pr.dfs(0,0);
  pr.print();
}