[BOJ] 14891 - 톱니바퀴

이준기·2022년 7월 23일
0

문제 링크

https://www.acmicpc.net/problem/14891

문제 설명

총 8개의 톱니를 가지고 있는 톱니바퀴 4개가 아래 그림과 같이 일렬로 놓여져 있다. 또, 톱니는 N극 또는 S극 중 하나를 나타내고 있다. 톱니바퀴에는 번호가 매겨져 있는데, 가장 왼쪽 톱니바퀴가 1번, 그 오른쪽은 2번, 그 오른쪽은 3번, 가장 오른쪽 톱니바퀴는 4번이다.

이때, 톱니바퀴를 총 K번 회전시키려고 한다. 톱니바퀴의 회전은 한 칸을 기준으로 한다. 회전은 시계 방향과 반시계 방향이 있다.

톱니바퀴를 회전시키려면, 회전시킬 톱니바퀴와 회전시킬 방향을 결정해야 한다. 톱니바퀴가 회전할 때, 서로 맞닿은 극에 따라서 옆에 있는 톱니바퀴를 회전시킬 수도 있고, 회전시키지 않을 수도 있다. 톱니바퀴 A를 회전할 때, 그 옆에 있는 톱니바퀴 B와 서로 맞닿은 톱니의 극이 다르다면, B는 A가 회전한 방향과 반대방향으로 회전하게 된다. 예를 들어, 아래와 같은 경우를 살펴보자.

사진이 많으므로 링크 참고 !

입력
첫째 줄에 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점

문제 풀이

주어진 시뮬레이션대로 구현하면 되는 문제다.

  • 처음에 톱니바퀴가 리스트인줄 알고 deque 등을 이용해서 풀었는데, 문자열이었다.. 더 간단
  • 양 쪽 톱니바퀴가 굴러가는 것을 결정할 때, 현재 톱니바퀴를 굴리기 전에 결정해야한다. 그래서 실제 굴리는 기능과 옆 톱니바퀴를 결정하는 기능을 나눴따.
    -> 시뮬레이션, 구현 문제는 모듈화를 잘하자 !
  • 톱니바퀴가 4개밖에 없으니까, 분기를 결정하는 방향을 직접 설정해서 moveDir로 부여했다.

맞은 코드

import copy

graph = []

for _ in range(4):
  graph.append(input())

def rotateSelf(num, dir):
  global graph
  tempGraph = copy.deepcopy(graph[num])
  graph[num] = ""
  if dir == -1: # 123456789
    graph[num] = tempGraph[1:8]
    graph[num] += tempGraph[0]
      
  else:  # 70123456
    graph[num] = ""
    graph[num] += tempGraph[7]
    for elem in tempGraph[0:7]:
      graph[num] += elem
    

def rotate(num, dir, moveDir):
  if moveDir == 1 and num < 3:
    if graph[num][2] != graph[num+1][6]:
      rotate(num+1, -1 * dir, 1)
  elif moveDir == -1 and num > 0:
    if graph[num][6] != graph[num-1][2]:
      rotate(num-1, -1 * dir, -1)
  elif moveDir == 0 and (num == 1 or num == 2):
    if graph[num][6] != graph[num-1][2]:
      rotate(num-1, -1 * dir, -1)
    if graph[num][2] != graph[num+1][6]:
      rotate(num+1, -1 * dir, 1)

  rotateSelf(num, dir)

k = int(input())
for _ in range(k): 
  # 방향이 1인 경우는 시계 -> 방향이고, -1인 경우는 반시계 <- 방향이다.
  wheelNum, wheelDir = map(int, input().split())
  if wheelNum == 1:
    moveDir = 1
  elif wheelNum == 4:
    moveDir = -1
  else:  # 0은 양방향
    moveDir = 0
  rotate(wheelNum-1, wheelDir, moveDir)
  #print(graph)

ans = 0
for idx, wheel in enumerate(graph):
  ans += (2**idx) * int(wheel[0])
print(ans)
profile
Hongik CE

0개의 댓글