백준: 15662 톱니바퀴 (2)

문제

15662번: 톱니바퀴 (2)

 

문제 풀이

 앞서 다룬 톱니바퀴 문제를 이해하고 있다면, 몇 가지 조건만 추가하면 해결할 수 있는 문제이다. 이전의 문제는 톱니바퀴의 개수가 4개로 고정되어 있었지만, 이 문제에서는 톱니바퀴의 수가 T로 입력으로 주어진다. 또한 출력은 K번 회전시킨 후 12시 방향이 S극인 톱니바퀴의 개수를 출력하는 것이다.

 

코드

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, t):
        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
    t = int(stdin.readline())
    gears = [deque(map(int, stdin.readline().strip())) for _ in range(t)]
    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([1 if gear[0] else 0 for delta, gear in enumerate(gears)]))