취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 9934번 문제인 완전 이진 트리이다.
문제는 아래 링크를 확인하자.

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

문제에서 완전이진트리의 inorder 순회를 입력으로 제공한다.

이 때, 완전이진트리의 각 깊이에 들어있는 노드의 수는 처음 주어진 수의 인덱스를 1로 잡고 이를 하나씩 올려나갈 때, 그 인덱스의 2진수표현의 끝에 0이 몇개 붙어있는가를 기준으로 판단할 수 있다.

순서대로 입력을 받아 인덱스의 0의 개수를 세서 벡터에 보관하고 마지막에 출력해주자.

아래는 제출한 소스코드이다.

#include <iostream>
#include <vector>
using namespace std;

vector<int> cat[11];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int N; cin >> N;
	int idx = 0;
	int x;
	while (cin >> x) {
		idx++;
		int temp = idx;
		int category = 0;
		while (!(temp & 1)) {
			category++;
			temp >>= 1;
		}
		cat[category].emplace_back(x);
	}
	for (int i = N - 1; i >= 0; i--) {
		for (auto x : cat[i]) cout << x << ' ';
		cout << '\n';
	}
}