백준 20055번: 컨베이어 벨트 위의 로봇

문제

문제 바로가기> 백준 20055번: 컨베이어 벨트 위의 로봇

풀이

로봇이 위치 N에서 내려지므로, 로봇은 컨베이어 벨트의 위에서만 움직인다.
간단한 구현문제였는데 이상하게 구현하다 헤맸다 ... ㅎ 현타온다 ㅎㅎㅎ
그래도 푸니까 신난당 ... ㅎㅎㅎㅎㅎㅎㅎㅎ

#include<iostream>
#include<vector>
#define MAX 201
using namespace std;

int N, K, ans=1;
int A[MAX]; // A[i] : 칸 i의 내구도
bool isrobot[MAX]; // 해당 칸의 로봇이 있는지 여부

int move(){
    
    // 1. 벨트가 각 칸 위에 있는 로봇과 함께 한 칸 회전
    int tmpA = A[2*N]; 
    bool tmpR = isrobot[2*N];
    for(int i=2*N; i>1; i--){
        A[i] = A[i-1];
        isrobot[i] = isrobot[i-1];
        if(i==N) isrobot[i] = false; // 언제든지 로봇이 내리는 위치에 도달하면 그 즉시 내림
    }
    A[1] = tmpA; 
    isrobot[1] = tmpR;

    // 2. 가장 먼저 벨트에 올라간 로봇부터, 벨트가 회전하는 방향으로 한 칸 이동할 수 있다면(내구도가 1 이상 남아 있는 경우) 이동
    for(int i=N-1; i>=0; i--){
        if(isrobot[i] && A[i+1]>0 && !isrobot[i+1]){
            if(i+1!=N) isrobot[i+1] = true;
            isrobot[i] = false;
            A[i+1]--;
        }
    }

    // 3. 올리는 위치에 있는 칸의 내구도가 0이 아니면 올리는 위치에 로봇을 올림
    if(A[1]>0) {
        isrobot[1] = true;
        A[1]--; // 로봇 올림, 내구도는 즉시 1만큼 감소
    }

    // 4. 내구도가 0인 칸의 개수가 K개 이상이라면 과정을 종료
    int durability=0;
    for(int i=1; i<=2*N; i++){
        if(A[i]==0) durability++;
    }
    return durability;
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);
    cin >> N >> K;
    for(int i=1; i<=2*N; i++) cin >> A[i];
    while(move()<K) ans++;
    cout << ans;
}