[C++]백준 20055번 컨베이어 벨트 위의 로봇

📌문제 링크

---

https://www.acmicpc.net/problem/20055

📌문제 설명

---

구현보다 이해가 어려운 문제입니다.

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

이 4가지 과정이 한단계로 각 과정을 그대로 구현하면 됩니다.

📌코드

---

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

int N, K;
vector<int> dur(201);
vector<bool> robot(201);
queue<int> robots;

// 입력을 받는 함수
void input() {
    cin >> N >> K;
    for (int i = 1; i <= 2 * N; i++) {
        cin >> dur[i];
    }
}

// 내구도가 0인 칸의 개수를 체크하는 함수
bool fourthQuery() {
    int cnt = 0;
    for (int i = 1; i <= 2 * N; i++) {
        if (dur[i] <= 0) cnt++;
    }
    return cnt >= K;
}

// 벨트와 로봇을 회전시키는 함수
void firstQuery() {
    int lastDur = dur[2 * N];
    for (int i = 2 * N; i > 1; i--) {
        dur[i] = dur[i - 1];
        robot[i] = robot[i - 1];
    }
    dur[1] = lastDur;
    robot[1] = false;

    // 내리는 위치(N)에서 로봇 제거
    robot[N] = false;
}

// 로봇을 이동시키는 함수
void secondQuery() {
    for (int i = N - 1; i >= 1; i--) {
        if (robot[i] && !robot[i + 1] && dur[i + 1] > 0) {
            robot[i] = false;
            robot[i + 1] = true;
            dur[i + 1]--;
        }
    }
    robot[N] = false; // 내리는 위치(N)에서 로봇 제거
}

// 새로운 로봇을 올리는 함수
void thirdQuery() {
    if (dur[1] > 0) {
        robot[1] = true;
        dur[1]--;
    }
}

// 문제를 해결하는 함수
void solve() {
    input();
    int cnt = 0;
    while (true) {
        cnt++;
        firstQuery();
        secondQuery();
        thirdQuery();
        if (fourthQuery()) {
            break;
        }
    }
    cout << cnt << '\n';
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
    return 0;
}