[BOJ] 공 (#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가 같을 수도 있다.

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

 

 

 

나의 코드 

t = int(input())
hash_map = {}

# 해시맵을 이용해 첫 번째 컵에 공(1)이 있도록 만든다. 
for i in range(1,4):
  if i == 1:
    hash_map[i] = 1
  else:
    hash_map[i] = 0

# 들어온 컵의 변화에 따라 value 값(공의 위치)가 바뀌도록 스위치해준다. 
for k in range(t):
  a,b = map(int,input().split())
  hash_map[b],hash_map[a] = hash_map[a],hash_map[b]

# 공의 위치를 key값으로 reverse하여 1이 어떤 위치에 있는지 알아낸다. 
reverse_dict= dict(map(reversed,hash_map.items()))
print(reverse_dict[1])

분석 > 나는 해시맵을 이용해서 문제를 풀었는데 우선 컵의 번호와 공의 위치를 연결한 해시맵을 만들었다. 그리고 컵의 변화에 따라 공의 위치가 바뀌도록 값을 스위치해주고 나중에 공의 위치를 key 값으로 보내 key 가 1인 경우의 value 값을 출력해서 문제를 풀었다.  이 코드를 작성하면서 해시맵에서 key 와 value 값을 바꿔주는 방법을 배울 수 있었다. 

 

 

1547번: 공