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

문제

N과 K가 주어지고,
1~N 까지는 로봇을 올릴 수 있고, N+1~2N 까지는 로봇을 내릴 수 있는 공간이다.

다음과 같은 작업이 반복된다.
1. 벨트가 각 칸 위에 있는 로봇과 함께 한 칸 회전한다.
2, 가장 먼저 벨트에 올라간 로봇부터, 벨트가 회전하는 방향으로 한 칸 이동할 수 있다면 이동한다. 만약 이동할 수 없다면 가만히 있는다.
3. 로봇이 이동하기 위해서는 이동하려는 칸에 로봇이 없으며, 그 칸의 내구도가 1 이상 남아 있어야 한다.
4, 올리는 위치에 있는 칸의 내구도가 0이 아니면 올리는 위치에 로봇을 올린다.
5, 내구도가 0인 칸의 개수가 K개 이상이라면 과정을 종료한다. 그렇지 않다면 1번으로 돌아간다.

내구도가 0인 갯수가 k개일 경우 반복횟수를 출력하는 문제다.

문제를 계속 읽어봤는데, 이해가 너무 안됐다.. 확실히 예제를 보고 그림을 그려가면서 문제를 이해하는게 빠르다는 것을 느꼈다 ㅠㅠ

[풀이 코드]

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    static int N, K, result;
    static int[] arr;
    static boolean[] robot;

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        N = Integer.parseInt(st.nextToken());
        K = Integer.parseInt(st.nextToken());
        arr = new int[2 * N];
        robot = new boolean[N];

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < 2 * N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        result = 0;

        while (true) {
            result++;

            // 벨트 및 로봇 회전
            move();

            // 로봇 이동
            moveRobots();

            // 로봇 올리기
            if (arr[0] > 0 && !robot[0]) {
                robot[0] = true;
                arr[0]--;
            }

            // 내구도 0인 칸 수 세기
            if (zeroCount() >= K) break;
        }

        System.out.println(result);
    }

    private static void move() {
        // 벨트 회전
        int last = arr[2 * N - 1];
        for (int i = 2 * N - 1; i > 0; i--) {
            arr[i] = arr[i - 1];
        }
        arr[0] = last;

        for (int i = N - 1; i > 0; i--) {
            robot[i] = robot[i - 1];
        }
        robot[0] = false;

        // 로봇이 내리는 위치에 도달하면 제거
        robot[N - 1] = false;
    }

    private static void moveRobots() {
        for (int i = N - 2; i >= 0; i--) {
            if (robot[i] && !robot[i + 1] && arr[i + 1] > 0) {
                robot[i] = false;
                robot[i + 1] = true;
                arr[i + 1]--;
            }
        }
        robot[N - 1] = false;
    }

    private static int zeroCount() {
        int count = 0;
        for (int a : arr) {
            if (a == 0) count++;
        }
        return count;
    }
}

일반적인 시뮬레이션 문제다.