[백준] 1057번 토너먼트

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

1057번: 토너먼트

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

vector<int> arr[18];

int main() {
	int n, a, b;
	cin >> n >> a >> b;
	for (int i = 1; i <= n; i++) {
		arr[0].push_back(i);
	}
	int row = 0;
	int result;
	bool flag = false;
	while (true) {
		for (int i = 0; i < arr[row].size(); i += 2) {
			if (i + 1 >= arr[row].size()) {
				arr[row + 1].push_back(arr[row][i]);
			}
			else if ((arr[row][i] == a || arr[row][i] == b) && (arr[row][i + 1] == a || arr[row][i + 1] == b)) {
				result = row + 1;
				flag = true;
				break;
			}
			else {
				if (arr[row][i] == a || arr[row][i] == b) {
					arr[row + 1].push_back(arr[row][i]);
				}
				else if (arr[row][i + 1] == a || arr[row][i + 1] == b) {
					arr[row + 1].push_back(arr[row][i + 1]);
				}
				else {
					arr[row + 1].push_back(arr[row][i]);
				}
			}
		}
		if (flag) break;
		row++;
	}
	cout << result;
	return 0;
}