[백준][1547] 공 | 파이썬

💡문제 분석 요약

세 개의 컵을 일렬로 엎어놓았다.

컵의 번호는 왼쪽 부터 1번, 2번, 3번 이다.

먼저 1번의 컵에 공을 하나 넣는다.

그 후 두 컵을 고른 다음 그 위치를 바꾸려고 한다. (공의 위치는 맨 처음 1번 컵이 있던 위치와 같다)

컵의 위치를 총 M번 바꿀 것이며, 컵의 위치를 바꾼 방법이 입력으로 주어진다.

위치를 M번 바꾼 이후에 공이 들어있는 컵의 번호를 구하는 프로그램을 작성하라.

💡알고리즘 설계

1. 교환 횟수를 입력받는다.
2. 공위치는 처음에 1번 컵에 있다는것을 설정해준다.
3. 공이 x번 컵에 있으면 y번 컵으로 이동하고, y번 컵에 있으면 x번 컵으로 이동시켜 준다.
4. 현재의 컵 번호를 출력해준다.

💡코드

m = int(input())  # 교환 횟수
ball_position = 1 # 공위치는 처음에 1번 컵에 있음음
# ball_position은 공이 현재 있는 컵 번호

for _ in range(m):
    x, y = map(int, input().split()) 
    if ball_position == x:  # 공이 x번 컵에 있으면
        ball_position = y  # y컵으로 이동
    elif ball_position == y:  # 공이 y번에 있으면
        ball_position = x    # x컵으로 이동

print(ball_position) # 결과 출력력

💡시간복잡도

map(int,input().split()) : 입력을 한 번 처리 O(1)

조건문 (if/ elif) : 공의 위치를 확인하고 필요시 업데이트 이는 O(1)

반복문은 총 m번 시행되고, 각 반복에서 실행되는 작업은 모두 상수시간O(1)이다.

O(1)

💡 틀린 이유

접근 방식이 틀림

💡 틀린 부분 수정 or 다른 풀이

💡 느낀점 or 기억할정보

생각보다 너무 복잡하게 생각했던 것 같다.