글 읽기 - (C++) 예제 4번이 해결되지 않습니다.

wlgur1023 3년 전 0

나름 직관적으로 논리를 구성했고, 딱히 틀린 부분은 없어 보이는데 예제 4번이 해결되지 않습니다.

정답은 472 인데 제 답은 497 이 나오더군요..

코드에 컨베이어 벨트의 상태를 보여줄 수 있도록 하는 프린트 함수를 추가하여 컨베이어 벨트가 돌아가는 과정을 직접 봐도 도무지 이상한 점을 찾을 수 없었습니다.

belt 구조체의 num 변수는 내구도를, isRobotOn 변수는 로봇이 벨트 위에 올라와 있는지 여부를 의미합니다.

어떤 부분에서 문제가 있었는지 확인해 주시면 감사하겠습니다.

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

int N, K;

typedef struct belt{
    int num;
    bool isRobotOn;
    belt() : num(0), isRobotOn(false) {}
}belt;

// 1. 벨트를 한 칸 회전한다.
//  * 내려가는 위치에 로봇이 있는 경우, 로봇을 내린다.
// 2. 로봇을 이동시킨다.
//  1) 가장 먼저 벨트에 올라간 로봇부터 한 칸 씩 이동한다.
//      이동 조건 : 이동하려는 칸에 로봇이 없으며, 그 칸의 내구도가 1 이상
//  * 내려가는 위치에 로봇이 있는 경우, 로봇을 내린다.
// 3. 올라가는 위치에 내구도가 0이 아니라면 로봇을 올린다.
// 4. 내구도가 0인 칸의 개수가 K개 이상이라면 과정을 종료한다. 그렇지 않다면 1번으로 돌아간다.

void Print(vector<vector<belt> > &belts){
    for(int y = 0; y < 2; ++y){
        for(int x = 0; x < N; ++x)
            cout << "(" << belts[y][x].num << ", " << belts[y][x].isRobotOn << ") ";
        cout << endl;
    }
}

void Input(vector<vector<belt> > &belts){
    cin >> N >> K;
    belts = vector<vector<belt> > (2, vector<belt> (N)); // zero base
    for(int y = 0; y < 2; ++y)
        for(int x = 0; x < N; ++x)
            cin >> belts[y][x].num;
}

void rotate_belt(vector<vector<belt> > &belts){
    // rotate belt
    belt buffer[2] = { belts[0][N - 1], belts[1][0] };

    for(int x = N - 1; x > 0; --x)
        belts[0][x] = belts[0][x - 1];
    for(int x = 0; x < N - 1; ++x)
        belts[1][x] = belts[1][x + 1];

    belts[1][N - 1] = buffer[0];
    belts[0][0]     = buffer[1];

    // unload robot at the unloading position.
    if(belts[0][N - 1].isRobotOn) belts[0][N - 1].isRobotOn = false;
}

void move_robots(vector<vector<belt> > &belts){    
    // position of belt "N"
    // unload robot at the unloading position.
    // if(belts[0][N - 1].isRobotOn) belts[0][N - 1].isRobotOn = false;
    
    // position of belt from "1" to "N-1"
    for(int x = N - 2; x >= 0; --x)
        if(belts[0][x].isRobotOn)
            if(!belts[0][x + 1].isRobotOn && belts[0][x + 1].num > 0){
                belts[0][x].isRobotOn     = false;
                belts[0][x + 1].isRobotOn = true;
                belts[0][x + 1].num--;
            }
            
    // check the position "N" again
    // unload robot at the unloading position.
    if(belts[0][N - 1].isRobotOn) belts[0][N - 1].isRobotOn = false;

}

bool loadRobot(vector<vector<belt> > &belts){
    if(belts[0][0].num > 0){
        belts[0][0].isRobotOn = true;
        belts[0][0].num--;
        return true;
    }
    else return false;
}

int countZeros(vector<vector<belt> > &belts){
    int zeros = 0;
    for(int y = 0; y < 2; ++y)
        for(int x = 0; x < N; ++x)
            if(belts[y][x].num == 0) ++zeros;
    return zeros;
}

int main(){
    vector<vector<belt> > belts;
    int cnt = 0;

    Input(belts);

        cout << "input            >> --------------------" << endl;
        Print(belts); cout << endl;

    while(countZeros(belts) < K){
        
        rotate_belt(belts);

            cout << "rotated belt >> --------------------" << endl;
            Print(belts); cout << endl;
        

        move_robots(belts);

            cout << "moved robot  >> --------------------" << endl;
            Print(belts); cout << endl;
        

        loadRobot(belts);

            cout << "loaded robot >> --------------------" << endl;
            Print(belts); cout << endl;
        
        ++cnt;
    }
    cout << cnt;
    
    return 0;
}

wlgur1023 3년 전 2

Input()을 엉뚱하게 받고 있었습니다..

컨베이어 벨트의 아랫줄 입력을 거꾸로 받고 있었습니다. 다음과 같이 Input() 함수를 고치니 바로 해결되었습니다.

void Input(vector<vector<belt> > &belts){
    cin >> N >> K;
    belts = vector<vector<belt> > (2, vector<belt> (N)); // zero base
        for(int x = 0; x < N; ++x)
            cin >> belts[0][x].num;
        for(int x = N - 1; x >= 0; --x)
            cin >> belts[1][x].num;
}

tori1753 1년 전 2

3시간동안 고민했는데 이글보고 해결했습니다

모니터부술뻔했는데 감사합니다

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

(C++) 예제 4번이 해결되지 않습니다.

wlgur1023 3년 전 0

wlgur1023 3년 전 2

tori1753 1년 전 2

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말