[백준 14891] 톱니바퀴 (Python)

👩🏻‍💻문제링크

[백준 14891] 톱니바퀴 (Python)

✍️Idea Sketch

2021-08-09

1. 문제 이해하기

• 톱니바퀴는 한 칸 이동을 기준으로, K번 회전한다.
• 시계 or 반시계방향으로 회전한다.
• A가 회전하기 전
  1. B와 극이 다른 경우, B는 A와 반대 방향으로 회전한다.
  2. B와 극이 같은 경우, B는 회전하지 않는다.

2. 입력값 살펴보기

• 12시부터 시계방향
  • 0은 N극, 1은 S극
• 회전 횟수 K
• 회전할 톱니바퀴 번호(num) 및 회전방향
  1. 1은 시계방향
  2. -1는 반시계방향

3. 재귀적 로직

• 예제1 참고) 톱니바퀴 3을 반시계방향으로 회전시켜보자.

1. 톱니바퀴 3과 오른쪽으로 인접한 톱니바퀴 4를 살펴보자. recursion()

   • graph[3][2]와 graph[4][6]의 두 극이
     1. 다른 경우, 톱니바퀴 4는 시계방향으로 회전한다. handle_rotation()
     2. 같은 경우, 톱니바퀴 4는 회전하지 않는다.
     • 톱니바퀴 4를 시계방향으로 회전시켜보자.

2. 톱니바퀴 3과 왼쪽으로 인접한 톱니바퀴 2를 살펴보자. recursion()

   • graph[3][6]과 graph[2][2]의 두 극이
     1. 다른 경우, 톱니바퀴 2는 시계방향으로 회전한다. handle_rotation()
     2. 같은 경우, 톱니바퀴 2는 회전하지 않는다.
     • 톱니바퀴 2를 시계방향으로 회전시켜보자.

✍️소스코드

2021-08-09 72ms 통과

import sys

def recursion(num, rotation, vs):  # 톱니바퀴 번호, 회전방향, 인접한 톱니바퀴 (1 or -1)
  if (0<num+vs<=4):
    if vs == 1:  # 오른쪽으로 인접한 톱니바퀴
      if graph[num][2] != graph[num+vs][6]:
        recursion(num+vs, -rotation, vs)
        handle_rotation(num+vs, -rotation)
    else:  # 왼쪽으로 인접한 톱니바퀴
      if graph[num][6] != graph[num+vs][2]:
        recursion(num+vs, -rotation, vs)
        handle_rotation(num+vs, -rotation)

def handle_rotation(num, rotation):
  if rotation == 1:  # 시계방향
    graph[num].insert(0, graph[num].pop())
  else:  # 반시계방향
    graph[num].append(graph[num][0])
    del graph[num][0]

if __name__ == "__main__":
  graph = [[0, 0, 0, 0, 0, 0, 0, 0]]
  for _ in range(4):
    graph.append(list(map(int, sys.stdin.readline().rstrip())))

  k = int(input())
  for _ in range(k):
    num, rotation = map(int, sys.stdin.readline().rstrip().split())
    recursion(num, rotation, 1)
    recursion(num, rotation, -1)
    handle_rotation(num, rotation)

  result = graph[1][0]
  result += graph[2][0]*2
  result += graph[3][0]*4
  result += graph[4][0]*8
  print(result)