[Baekjoon] 14891_톱니바퀴

Gold V

문제(출처: https://www.acmicpc.net/problem/14891)

---

<톱니바퀴>

문제 

 

* 문제가 길어서 생략

 

 

입력

 

첫째 줄에 1번 톱니바퀴의 상태, 둘째 줄에 2번 톱니바퀴의 상태, 셋째 줄에 3번 톱니바퀴의 상태, 넷째 줄에 4번 톱니바퀴의 상태가 주어진다. 상태는 8개의 정수로 이루어져 있고, 12시 방향부터 시계방향 순서대로 주어진다. N극은 0, S극은 1로 나타나 있다.

다섯째 줄에는 회전 횟수 K(1 ≤ K ≤ 100)가 주어진다. 다음 K개 줄에는 회전시킨 방법이 순서대로 주어진다. 각 방법은 두 개의 정수로 이루어져 있고, 첫 번째 정수는 회전시킨 톱니바퀴의 번호, 두 번째 정수는 방향이다. 방향이 1인 경우는 시계 방향이고, -1인 경우는 반시계 방향이다.

 

출력 

 

총 K번 회전시킨 이후에 네 톱니바퀴의 점수의 합을 출력한다. 점수란 다음과 같이 계산한다.

1번 톱니바퀴의 12시 방향이 N극이면 0점, S극이면 1점
2번 톱니바퀴의 12시 방향이 N극이면 0점, S극이면 2점
3번 톱니바퀴의 12시 방향이 N극이면 0점, S극이면 4점
4번 톱니바퀴의 12시 방향이 N극이면 0점, S극이면 8점

 

 

문제 풀이

 

 - my solution (Python)

 

import sys
from collections import deque

def change( direction, num):
    before = arr[num]
    if direction==1: # 시계
        before=list(before[-1])+before[:7]
    else: # 반시계
        before=before[1::]+list(before[0])

    return before

arr=[] # 톱니바퀴 상태
temp_arr=[] # 톱니바퀴 돌아가는 상태
for i in range(4): 
    x=list(sys.stdin.readline().strip())
    arr.append(x)
    temp_arr.append(x)

k=int(input()) # 회전 횟수
command=deque([]) # 회전시킨 방법
for i in range(k):
    command.append(list(map(int,sys.stdin.readline().split())))

for i in command:
    visited=[0,0,0,0] # 방문 여부
    ex_command=deque([i])
    visited[i[0]-1]=1
    while ex_command:
        temp=ex_command.popleft()
        temp_arr[temp[0]-1]=change(temp[1], temp[0] - 1) 
        if temp[0]==1: # 1번
            if arr[0][2]!=arr[1][6]: 
                if temp[1]==1: # 시계
                        temp_arr[1]=change(-1, 1)
                        if visited[1]==0:
                            visited[1]=1
                            ex_command.appendleft([2,-1])
                else: # 반시계
                    temp_arr[1]=change(1,1)
                    if visited[1] == 0:
                        visited[1] = 1
                        ex_command.appendleft([2,1])
        elif temp[0]==4: # 4번
            if arr[3][6]!=arr[2][2]: 
                if temp[1]==1: # 시계
                    temp_arr[2]=change(-1,2)
                    if visited[2] == 0:
                        visited[2] = 1
                        ex_command.appendleft([3, -1])
                else: # 반시계
                    temp_arr[2]=change(1,2)
                    if visited[2] == 0:
                        visited[2] = 1
                        ex_command.appendleft([3, 1])
        else: # 2번, 3번 
            if arr[temp[0]-1][2]!=arr[temp[0]][6]:
                if temp[1]==1: # 시계
                    temp_arr[temp[0]]=change(-1, temp[0])
                    if visited[temp[0]] == 0:
                        visited[temp[0]] = 1
                        ex_command.appendleft([temp[0]+1, -1])
                else: # 반시계
                    temp_arr[temp[0]] = change(1,temp[0])
                    if visited[temp[0]] == 0:
                        visited[temp[0]] = 1
                        ex_command.appendleft([temp[0]+1, 1])
            if arr[temp[0]-1][6]!=arr[temp[0]-2][2]:
                if temp[1]==1: # 시계
                    temp_arr[temp[0]-2]=change(-1,temp[0]-2)
                    if visited[temp[0]-2] == 0:
                        visited[temp[0]-2] = 1
                        ex_command.appendleft([temp[0]-1, -1])
                else: # 반시계
                    temp_arr[temp[0]-2]=change(1, temp[0]-2)
                    if visited[temp[0]-2] == 0:
                        visited[temp[0]-2] = 1
                        ex_command.appendleft([temp[0]-1, 1])
    arr=temp_arr[:]

result=0
if arr[0][0]=='1': # 1번
    result+=1
if arr[1][0]=='1': # 2번
    result+=2
if arr[2][0]=='1': # 3번
    result+=4
if arr[3][0]=='1': # 4번
    result+=8

print(result)

 

문제부터 이해하는데 오래 걸렸다. 

 

❓ 어느 부분을 이해하는데 오래 걸렸는가?

톱니바퀴가 돌아가는데 원래 상태에서 극이 같은지 판별하는 것 때문이었다. 처음에 내가 생각하기에는 하나의 톱니바퀴가 돌아간 후 그 상태에서 연속적으로 일어나는 일인 줄 알았기 때문이었다.

 

 

• change ( 돌아가는 방향, 톱니바퀴 번호 )
  : 시계 방향, 반시계 방향
• arr = 톱니바퀴 상태
  temp_arr = 톱니바퀴가 여러 개 돌아갈 때 상태
  k = 회전 횟수 입력
  command = 회전시킨 방법 입력
  visited = 방문 여부
  ex_command = [맞닿은 톱니의 극이 다른 톱니바퀴 번호, 돌아가는 방향] 저장
• command 순환 
  = ex_command가 빌 때까지 ( 톱니바퀴가 더 이상 돌아가지 않을 때까지 ) -> ex_command 맨 앞 값 pop -> 돌아가야 하는 방향으로 돌림 -> 각 톱니바퀴의 맞닿은 부분이 다른 극이라면 ex_command에 넣어줌
• 점수 계산 및 출력

# 맞닿은 톱니의 극이 다르면 맞닿은 톱니바퀴가 돌아가야 하고, 그 톱니바퀴와 맞닿은 톱니의 극이 다르면.... 이런 식으로 반복되어야 하므로 temp_arr을 만들어 톱니바퀴가 돌아가는 중에 바로 상태가 변하지 않도록 해주었다. 

 

# 1번 톱니바퀴는 2번이랑만 맞닿으므로 1번의 3번 톱니 - 2번의 7번 톱니를 비교해준다.

마찬가지로 4번 톱니바퀴는 3번이랑만 맞닿으므로 4번의 7번 - 3번의 3번을 비교해준다.

2번 톱니바퀴와 3번 톱니바퀴는 양 옆으로 맞닿으므로 왼쪽, 오른쪽 톱니바퀴와 비교해준다.

 

 

❓ 처음에 왜 틀렸는가?

와악-! 다 풀고 여기서 시간이 걸렸다. 주어진 예제를 다 통과해서 제출했더니 바로 틀렸습니다가 떴다. 바로 틀렸습니다가 발생하여 구현 자체가 틀린 줄 알고 다른 사람들의 질문과 반레를 다 찾아보았지만 틀린 부분을 찾을 수 없었다. 다시 처음부터 코드를 살펴보니 ex_command에 회전 방향을 하나 반대로 입력하여 틀렸던 것이었다. ^^

주어진 예제에서는 그 코드를 지나가는 부분이 없었나 보다... 예제 다 통과한 거 보니... 찾아서 다행이다 

 

---

생각🤔

 

코드가 너무 길다는 생각이 들지만... 문제 이해와 통과한 것만으로도 내 머리는 용량 초과이다...

 

---