BOJ 20055 - 컨베이어 벨트 위의 로봇

문제

백준 온라인 저지 - 20055번

풀이 과정

문제에서 제시한 규칙대로 동작하는 컨베이어 벨트를 구현하면 되는 문제입니다.

2차원 배열을 이용해서 컨베이어 벨트를 나타냈지만 사실 그냥 1차원 배열을 이용해서 구현해도 될 것 같습니다.
문제에서 주어진 네 가지 단계를 각각의 함수로 구현해서 검사해주는 방식으로 구현했습니다.

코드

import sys
from collections import deque

N, K = list(map(int, sys.stdin.readline().split()))
belts = list(map(int, sys.stdin.readline().split()))


def init_conveyor():
    conveyor = [[0] * N for _ in range(2)]
    conveyor[0] = [*belts[:N]]
    conveyor[1] = [*belts[N:][::-1]]

    return conveyor


def rotate_belt(conveyor, robots):
    # 컨베이어 벨트 이동
    cache1 = conveyor[0][N - 1]
    for y in range(N - 1, 0, -1):
        conveyor[0][y] = conveyor[0][y - 1]

    cache2 = conveyor[1][0]
    for y in range(N - 1):
        conveyor[1][y] = conveyor[1][y + 1]

    conveyor[0][0] = cache2
    conveyor[1][N - 1] = cache1

    # 로봇 이동
    robots.pop()
    robots.appendleft(0)


def move_robots(conveyor, robots):
    for idx in range(N - 1, -1, -1):
        if robots[idx]:
            next_idx = idx + 1
            if robots[next_idx] == 0 and conveyor[0][next_idx] > 0:
                conveyor[0][next_idx] -= 1
                robots[idx] = 0
                robots[next_idx] = 1


def add_robot(conveyor, robots):
    if conveyor[0][0] > 0 and robots[0] != 1:
        robots[0] = 1
        conveyor[0][0] -= 1


def check_belt(conveyor):
    cnt = 0

    for x in range(2):
        for y in range(N):
            if not conveyor[x][y]:
                cnt += 1

    return cnt < K


def check_endpoint(robots):
    if robots[N - 1]:
        robots[N - 1] = 0


def solution():
    answer = 1
    conveyor = init_conveyor()
    robots = deque([0] * N)

    while True:
        rotate_belt(conveyor, robots)
        check_endpoint(robots)
        move_robots(conveyor, robots)
        check_endpoint(robots)
        add_robot(conveyor, robots)

        go = check_belt(conveyor)
        if not go:
            break

        answer += 1

    return answer


print(solution())