△△ 게임대회가 개최되었습니다. 이 대회는 N명이 참가하고, 토너먼트 형식으로 진행됩니다. N명의 참가자는 각각 1부터 N번을 차례대로 배정받습니다. 그리고, 1번↔2번, 3번↔4번, ... , N-1번↔N번의 참가자끼리 게임을 진행합니다. 각 게임에서 이긴 사람은 다음 라운드에 진출할 수 있습니다. 이때, 다음 라운드에 진출할 참가자의 번호는 다시 1번부터 N/2번을 차례대로 배정받습니다. 만약 1번↔2번 끼리 겨루는 게임에서 2번이 승리했다면 다음 라운드에서 1번을 부여받고, 3번↔4번에서 겨루는 게임에서 3번이 승리했다면 다음 라운드에서 2번을 부여받게 됩니다. 게임은 최종 한 명이 남을 때까지 진행됩니다.
이때, 처음 라운드에서 A번을 가진 참가자는 경쟁자로 생각하는 B번 참가자와 몇 번째 라운드에서 만나는지 궁금해졌습니다. 게임 참가자 수 N, 참가자 번호 A, 경쟁자 번호 B가 함수 solution의 매개변수로 주어질 때, 처음 라운드에서 A번을 가진 참가자는 경쟁자로 생각하는 B번 참가자와 몇 번째 라운드에서 만나는지 return 하는 solution 함수를 완성해 주세요. 단, A번 참가자와 B번 참가자는 서로 붙게 되기 전까지 항상 이긴다고 가정합니다.
N | A |
---|---|
8 | 4 |
첫 번째 라운드에서 4번 참가자는 3번 참가자와 붙게 되고, 7번 참가자는 8번 참가자와 붙게 됩니다. 항상 이긴다고 가정했으므로 4번 참가자는 다음 라운드에서 2번이 되고, 7번 참가자는 4번이 됩니다. 두 번째 라운드에서 2번은 1번과 붙게 되고, 4번은 3번과 붙게 됩니다. 항상 이긴다고 가정했으므로 2번은 다음 라운드에서 1번이 되고, 4번은 2번이 됩니다. 세 번째 라운드에서 1번과 2번으로 두 참가자가 붙게 되므로 3을 return 하면 됩니다.
테스트 케이스에서 틀린 풀이
import copy
def solution(n,a,b):
answer = 0
part=[i for i in range(n)]
for j in range(n-1):
answer+=1
win=[]
for i in range(0,n-1,2):
if a-1 in [part[i],part[i+1]] and b-1 in [part[i],part[i+1]]:
return answer
if a-1 in [part[i],part[i+1]]:
win.append(a-1)
elif b-1 in [part[i],part[i+1]]:
win.append(b-1)
else:
win.append(i)
n//=2
part=copy.deepcopy(win)
return answer
도대체 어디서 틀린건지 감을 못 잡다가 win.append(i)에서 i는 인덱스일뿐 값이 아니라는 것을 깨닫고 다시 코드를 짰다.
성공한 풀이
import copy
def solution(n,a,b):
answer = 0
part=[i for i in range(1,n+1)]
while part:
win=[]
answer+=1
for i in range(0,n,2):
if [a,b] == [part[i],part[i+1]] or [b,a] == [part[i],part[i+1]]:
return answer
if a in [part[i],part[i+1]]:
win.append(a)
elif b in [part[i],part[i+1]]:
win.append(b)
else:
win.append(part[i])
n//=2
part=copy.deepcopy(win)
return answer+1
from collections import deque
def solution(n,a,b):
answer = 0
q = deque([i for i in range(1,n+1)])
while(q):
x = q.popleft()
y = q.popleft()
if [x,y] == [a,b] or [x,y] == [b,a]:
answer+=1
break
if a in [x, y]: # list를 쓰면 한 줄로 item 검색이 가능
q.append(a)
answer+=1
elif b in [x, y]: # list를 쓰면 한 줄로 item 검색이 가능
q.append(b)
else:
q.append(x)
return answer