[백준] 20361번 일우는 야바위꾼 파이썬

20361번 일우는 야바위꾼 파이썬 풀이

 

[시도1] pypy3 시간 초과

n, x, k = map(int, input().split())

turn = [list(map(int, input().split())) for _ in range(k)]

cup = [0] * n
cup[x-1] += 1 

for i in range(0, k):
  for j in range(0, 2):
    if turn[i][j] == cup.index(1)+1:
      if j == 0:
        cup[turn[i][j]-1] -= 1
        cup[turn[i][j+1]-1] += 1
        break
      else:
          cup[turn[i][j]-1] -= 1
          cup[turn[i][j-1]-1] += 1
      
print(cup.index(1)+1)

 

=> 좀 더 효율적으로 코드를 작성할 수 있는 방법을 고민해봐야 한다.

 

www.acmicpc.net/problem/20361