[백준14891&파이썬] 톱니바퀴를 돌릴때는 deque를 사용하자

# 문제

백준 14891 파이썬 풀이

14891번: 톱니바퀴

# 코드

import sys
from collections import deque

LEFT = -1
RIGHT = 1

# N극은 0, S극은 1
A = deque(list(map(int, sys.stdin.readline().strip())))
B = deque(list(map(int, sys.stdin.readline().strip())))
C = deque(list(map(int, sys.stdin.readline().strip())))
D = deque(list(map(int, sys.stdin.readline().strip())))

arrs = [[], A, B, C, D]

# 회전 횟수
K = int(sys.stdin.readline())

def rotate(idx, dir):
  global arrs, LEFT, RIGHT
  if dir == LEFT:
    arrs[idx].append(arrs[idx].popleft())
  if dir == RIGHT:
    arrs[idx].appendleft(arrs[idx].pop())


for _ in range(K):
  idx, dir = map(int, sys.stdin.readline().split())

  # bfs처럼 양옆 톱니들을 비교해야한다. (전파)
  dq =deque()
  dq.append((idx,dir)) # 톱니, 회전방향( 반대방향으로 회전시켜야 하기 때문에 )
  visit = [False] * 5
  visit[idx] = True
  l = deque()
  r = deque()
  while len(dq) !=0 :
    midx, mdir = dq.pop()
    nl = midx - 1; nr = midx + 1  # 양옆 톱니

    if 1 <= nl <= 3 and visit[nl] == False:
      if arrs[midx][6] != arrs[nl][2] :
        # rotate(nl,-1*mdir) # 반대방향으로 회전시킨다.
        visit[nl] = True # 방문처리
        dq.append((nl,-1*mdir))
        l.append((nl,-1*mdir))

    if 2 <= nr <= 4 and visit[nr] == False:
      if arrs[midx][2] != arrs[nr][6]:
        # rotate(nr, -1*mdir)  # 반대방향으로 회전시킨다.
        visit[nr] = True  # 방문처리
        dq.append((nr, -1*mdir))
        r.append((nr, -1*mdir))

  for i in range(len(l)) :
    midx, mdir = l.pop()
    rotate(midx,mdir)
  for i in range(len(r)) :
    midx,mdir = r.pop()
    rotate(midx,mdir)
  rotate(idx, dir) # 톱니 회전
  # print(arrs)



# N극은 0, S극은 1
# print(arrs)

result = 0
for idx in range(1,5) :
  if arrs[idx][0] == 1 :
    result += pow(2,idx-1)

print(result)

# 풀이

• 톱니바퀴를 회전시키는 문제이다. 회전문제는 모듈러 연산이나 deque를 이용해서 돌리면 된다.
• 다 풀고나서 알았는데 deque자체에 rotate()라는 함수를 지원한다. 다음에는 이걸 사용해서 풀어봐야겠다.
• 난 아직도 이문제를 풀고나서 명확히 이해가 되지 않는다.
• 내가 문제를 이해하기로는 현재 입력받은 톱니를 돌리고 나서 양 극관계를 BFS로 파악해 나가는것인줄 알았다. 그래서 문제도 deque를 사용해서 BFS로 풀었다.
• 근데 다 풀고나니까 현재 상태에서 극을 보고 먼저 판단을 해야한다. DFS느낌으로 맨끝에를 돌리고나서 현재로 돌아와 입력받은거를 돌려야 했다. 
• 근데 이건 문제의 어패가 있는거 같다. 이 방식이라면 최종적으로 현재를 rotate시켰을때 양옆의 값이 다시 N/S극이라면 또 돌려야 하는거 아닌가? 마치 무한동력처럼 계속 돌아가야하는거 같다..
• 여튼 문제 해결을 위해서는 어찌저찌 풀었는데, 내 이해력이 약한건지는 몰라도 문제에 대한 이런 부가설명이 조금 더 있었으면 좋겠다.
• (그치만 SW기출이기때문에 내가 이해를 못한거다..라고 하자)