백준: 14891 톱니바퀴

문제

14891번: 톱니바퀴

 

문제 풀이

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

 문제에는 4개의 톱니바퀴에 대한 상태가  N극인 경우 0, S극인 경우 1로 주어진다. 하나의 톱니바퀴를 회전할 때, 시계방향 또는 반시계 방향으로 회전할 수 있고 회전하고자 하는 톱니바퀴와 인접한 톱니바퀴는 극이 다른 경우 영향을 받게 된다. 이때, 회전한 톱니바퀴와는 반대 반향으로 회전하는 특징을 가지고 있다. 이를 고려하여 로직을 구현하면 쉽게 해결할 수 있는 문제이다.

 

 하나의 톱니바퀴를 회전시키면 좌측, 우측에 연쇄적으로 작용한다. 따라서 선택된 톱니바퀴의 좌우측의 톱니바퀴들이 회전한다면, 회전한 톱니바퀴의 좌측 또는 우측도 모두 확인하여야 한다.

 

코드

from sys import stdin
from collections import deque


def move_gear(idx, cur_dir):
    global cur_right, cur_left, gears
    origin_dir = cur_dir

    # 현재 톱니바퀴를 기준으로 좌측의 톱니바퀴들 확인.
    for i in reversed(range(idx)):
        if cur_right != gears[i][LEFT]:
            cur_right = gears[i][RIGHT]
            gears[i].rotate(cur_dir * -1)
            cur_dir *= -1
        else:
            break

    cur_dir = origin_dir
    # 현재 톱니바퀴를 기준으로 우측의 톱니바퀴들 확인.
    for i in range(idx + 1, 4):
        if gears[i][RIGHT] != cur_left:
            cur_left = gears[i][LEFT]
            gears[i].rotate(cur_dir * -1)
            cur_dir *= -1
        else:
            break


if __name__ == '__main__':
    LEFT, RIGHT = 2, 6
    gears = [deque(map(int, stdin.readline().strip())) for _ in range(4)]
    k = int(stdin.readline())
    orders = [list(map(int, stdin.readline().split())) for _ in range(k)]

    for gear_idx, direction in orders:
        gear_idx -= 1
        cur_left, cur_right = gears[gear_idx][LEFT], gears[gear_idx][RIGHT]
        gears[gear_idx].rotate(direction)
        move_gear(gear_idx, direction)

    print(sum([2 ** delta if gear[0] else 0 for delta, gear in enumerate(gears)]))

 처음에는 rotation하는 함수를 작성하였는데, deque를 활용하면 rotate함수가 있어 이를 구현할 필요가 없다는 걸 처음 알았다.😂 반복문을 통해 연쇄적으로 영향을 미치는 톱니바퀴까지 회전하게 되면 정답이라는 문구를 볼 수 있다.