[Python] 백준 1057 - 토너먼트

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

1057번: 토너먼트

 

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 수학
 - 브루트포스 알고리즘

문제

김지민은 N명이 참가하는 스타 토너먼트에 진출했다. 토너먼트는 다음과 같이 진행된다. 일단 N명의 참가자는 번호가 1번부터 N번까지 배정받는다. 그러고 난 후에 서로 인접한 번호끼리 스타를 한다. 이긴 사람은 다음 라운드에 진출하고, 진 사람은 그 라운드에서 떨어진다. 만약 그 라운드의 참가자가 홀수명이라면, 마지막 번호를 가진 참가자는 다음 라운드로 자동 진출한다. 다음 라운드에선 다시 참가자의 번호를 1번부터 매긴다. 이때, 번호를 매기는 순서는 처음 번호의 순서를 유지하면서 1번부터 매긴다. 이 말은 1번과 2번이 스타를 해서 1번이 진출하고, 3번과 4번이 스타를 해서 4번이 진출했다면, 4번은 다음 라운드에서 번호 2번을 배정받는다. 번호를 다시 배정받은 후에 한 명만 남을 때까지 라운드를 계속 한다.

마침 이 스타 대회에 임한수도 참가했다. 김지민은 갑자기 스타 대회에서 우승하는 욕심은 없어지고, 몇 라운드에서 임한수와 대결하는지 궁금해졌다. 일단 김지민과 임한수는 서로 대결하기 전까지 항상 이긴다고 가정한다. 1 라운드에서 김지민의 번호와 임한수의 번호가 주어질 때, 과연 김지민과 임한수가 몇 라운드에서 대결하는지 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 참가자의 수 N과 1 라운드에서 김지민의 번호와 임한수의 번호가 순서대로 주어진다. N은 2보다 크거나 같고, 100,000보다 작거나 같은 자연수이고, 김지민의 번호와 임한수의 번호는 N보다 작거나 같은 자연수이고, 서로 다르다.

출력

첫째 줄에 김지민과 임한수가 대결하는 라운드 번호를 출력한다. 만약 서로 대결하지 않을 때는 -1을 출력한다.

 

---

 

간만에 생각하기 좋은 문제였다.

 

import sys
import math
input = sys.stdin.readline

total, a, b = map(int, input().split())

lvl = 1
while True:
    if (a%2==1 and b%2==0 and b-a == 1) or (a%2==0 and b%2==1 and a-b==1):
        print(lvl)
        break
    a = int(math.ceil(a/2))
    b = int(math.ceil(b/2))
    lvl += 1

처음에는 조건식에

 

if a%2==1 and b%2==0 and b-a == 1:

이것만 넣었었다. 하지만 시간초과가 발생했고

 

곰곰이 생각해 보니 반드시 '김지민'의 번호가 '임한수'의 번호보다 앞선 번호라는 보장이 없었다.

 

그래서 반대의 경우도 추가해 주니 해결되었다.

 

문제 출력에 나온 '서로 대결하지 않을 경우 -1 출력'은 고려하지 않아도 된다.

 

둘 다 무조건 서로 만날 때까지 전부 토너먼트에 진출한다 하였고 토너먼트에서는 결국 둘이 만날 수밖에 없기 때문이다.