백준 1547번 파이썬 풀이 : 공

문제 링크 : https://www.acmicpc.net/problem/1547

1547번: 공

문제

세준이는 컵 3개를 탁자 위에 일렬로 엎어놓았다. 컵의 번호는 맨 왼쪽 컵부터 순서대로 1번, 2번 3번이고, 세준이는 이 컵을 이용해서 게임을 하려고 한다.

먼저 1번 컵의 아래에 공을 하나 넣는다. 세준이는 두 컵을 고른 다음, 그 위치를 맞바꾸려고 한다. 예를 들어, 고른 컵이 1번과 2번이라면, 1번 컵이 있던 위치에 2번 컵을 이동시키고, 동시에 2번 컵이 있던 위치에 1번 컵을 이동시켜야 한다. 이때 공은 움직이지 않기 때문에, 공의 위치는 맨 처음 1번 컵이 있던 위치와 같다.

세준이는 컵의 위치를 총 M번 바꿀 것이며, 컵의 위치를 바꾼 방법이 입력으로 주어진다. 위치를 M번 바꾼 이후에 공이 들어있는 컵의 번호를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 컵의 위치를 바꾼 횟수 M이 주어지며, M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 M개의 줄에는 컵의 위치를 바꾼 방법 X와 Y가 주어지며, X번 컵과 Y번 컵의 위치를 서로 바꾸는 것을 의미한다. X와 Y의 값은 3보다 작거나 같고, X와 Y가 같을 수도 있다.

컵을 이동시킨 후에 공이 컵 바깥에 있는 경우는 없다.

출력

첫째 줄에 공이 들어있는 컵의 번호를 출력한다. 공이 사라져서 컵 밑에 없는 경우에는 -1을 출력한다.

 

문제 풀이

# 공의 갯수는 3개 고정이다.

# 공의 위치는 움직이지 않는다.

# 이 두가지의 문제를 가지고 문제를 풀이했다. 

 

1. 3칸 크기의 배열을 생성한 후 1, 2, 3을 대입한다.

2. 예제를 입력하면 해당 값이 들어있는 배열의 값들을 바꿔 준다. ex) 3 1 을 입력하면 배열에는 3, 2, 1이 들어간다.

3. 0번째 배열의 값을 출력해주면 공이 들어있는 컵의 번호를 알 수 있다.

 

# 여기서 중요한 점은 예제로 입력하는 값이 배열의 인덱스가 아니라 배열의 값 이기 때문에 값을 입력하면 배열의 인덱스를 리턴해주는 함수를 새로 구현해 주었다.

* 이 부분도 배열이 3칸으로 고정이 되어있어 가능하다는 점, 컵의 갯수가 많아지면 코드의 효율성도 떨어진다.

def getIndex(num, set_answer):
    for i in range(3):
        if set_answer[i] == num:
            return i
 
M = int(input())
answer = [1, 2, 3]
tempX, tempY = 0, 0
for i in range(M):
    X, Y = map(int, input().split())
    tempX = getIndex(X, answer)
    tempY = getIndex(Y, answer)
    temp = answer[tempX]
    answer[tempX] = answer[tempY]
    answer[tempY] = temp
print(answer[0])
 

컵의 위치는 움직이지 않기 때문에 answer 배열의 첫번째인 0번째 배열을 출력해주면 해당 공이 있는 컵의 번호를 알 수 있다.

 

전체 코드 링크 : https://github.com/Hmgi/Prepare-CodingTest/blob/master/Baekjoon/1547.py