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

처음 문제를 읽는데 문제 이해가 헷갈렸다.

문제를 해석하면 컨베이어 벨트는 시계 방향으로 원을 그리며 회전한다.

 

컨베이어 벨트가 회전할 때 위에 있는 로봇은 같이 움직이고, 벨트가 움직인 후 다시 로봇이 오른쪽으로 조건에 맞게 이동이 가능하다!!

 

그 후, 로봇이 올라가는 위치에서 올라갈 수 있다.

 

처음부터 로봇이 올라가는 것이 아니라 1단계에서는 로봇이 없는 채로 1번부터 4번까지 진행을 한다.

즉, 1,2번 진행 후 첫 로봇이 3번째에 올라가게 되는 것이다.

 

위의 문제에서 제시된 과정을 순서대로 작성하고 실행하면 된다.

 

package algo;
 
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class B_20055_컨베이어벨트위의로봇 {
 
    static int cnt;
 
    public static void main(String[] args) throws Exception {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(in.readLine(), " ");
        int N = Integer.parseInt(st.nextToken());
        int K = Integer.parseInt(st.nextToken());
        int M = 2 * N;
        int[] arr = new int[M];
        boolean[] robot = new boolean[N];
 
        st = new StringTokenizer(in.readLine(), " ");
        for (int i = 0; i < M; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
 
        int time = 0; // 시작 - 1단계
 
        while (cnt < K) {
            time++; // 1단계
            rotate(arr, M, N, robot); // 1. 벨트 회전
            go(robot, arr, N); // 2. 로봇 이동
            if (arr[0] > 0) { // 3. 로봇 올리기 - 올리는 위치
                arr[0] -= 1;
                if (arr[0] == 0)
                    cnt++;
                robot[0] = true;
            }
        }
        System.out.println(time); // 결과 출력
    }
 
    // 로봇 이동
    private static void go(boolean[] robot, int[] arr, int N) {
        for (int i = N - 1; i >= 0; i--) { // 로봇이 먼저 올라간 순서로
            if (i == N - 1) { // 내리는 위치
                robot[N - 1] = false;
                continue;
            }
            if (robot[i]) { // 로봇이 있다면
                if (!robot[i + 1] && arr[i + 1] >= 1) { // 2-1.이동이 가능하면
                    arr[i + 1] -= 1; // 내구도 감소
                    if (arr[i + 1] == 0)
                        cnt++;
                    robot[i] = false;
                    robot[i + 1] = true; // 이동
                }
            }
        }
    }
 
    // 각 벨트의 회전
    private static void rotate(int[] arr, int M, int N, boolean[] robot) {
        int temp = arr[M - 1];
        for (int i = M - 2; i >= 0; i--) {
            arr[i + 1] = arr[i];
        }
        arr[0] = temp;
        // 벨트 이동과 함께 로봇 이동
        for (int i = N - 2; i >= 0; i--) {
            robot[i + 1] = robot[i];
        }
        robot[0] = false;
    }
}