[백준/파이썬] 1547번 공 풀이

업데이트:



문제 정보


풀이

문제

1번 컵 아래에 공이 있고, 여러 번 컵 위치를 바꾼 뒤 공이 있는 컵 번호를 출력하는 문제입니다.

코드

import sys

cups = [0]*4
cups[1] = 1

for _ in range(int(sys.stdin.readline())):
    a, b = map(int, sys.stdin.readline().split())
    cups[a], cups[b] = cups[b], cups[a]

for i in range(1, 4):
    if cups[i] == 1:
        print(i)
        break
    if i == 3:
        print(-1)

설명

컵 상태 배열을 두고 입력된 두 컵 번호를 매번 스왑합니다.

모든 교환 후 공(1)이 있는 컵 인덱스를 출력합니다.



댓글남기기