글 읽기 - 14503 로봇 청소기문제 질문드립니다...

peach404 5년 전 0

안녕하세요... 제가 로직을 잘못 생각한건지.. 테스트 케이스는 잘 돌아가는데 정답이 틀렸다고 나옵니다.

도와주세요.....

#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <functional>
#include <algorithm>

using namespace std;

queue<pair<int, int> > q;
int arr[51][51];
int visited[51][51];

int bx[4] = {-1, 0, 1, 0};
int by[4] = {0, 1, 0, -1};

int X, Y;
int ry, rx, d;
int cnt = 1;
int nx, ny;
int od;
int iny, inx;


bool icheck(int i) {
    if (i == 3)return true;
    else return false;
}

void bfs() {

    while (!q.empty()) {
        iny = q.front().first;
        inx = q.front().second;

        q.pop();


        for (int k = 1; k <= 5; k++) {

            int i=d;

            ny = iny + by[i];
            nx = inx + bx[i];
            
            // 범위 안에 해당하는 값인지 && 청소안한 곳인지 && 벽이 아닌지 체크합니다.
            if (ny>=0 &&  nx>=0 &&nx<X && ny<Y && visited[ny][nx] == 0 && arr[ny][nx] == 0) { 
                q.push({ny, nx});
                visited[ny][nx] = 1;
                cnt++; //청소한 공간 세는 변수입니다.
                od = d; //for문 돌면서 d의 값이 변경되기 때문에 후진여부를 체크할 때 필요한 변수입니다.
                if (icheck(d)) { d = 0; } else { d++; } //i 값이 3이면 다시 d를 0으로 변경하여 검사를 시작합니다. 그 이외의 값은 1씩 증가시켜 검사합니다.

                break;
            }

            if (icheck(i)) { //청소 기준에 맞지 않는 곳도 d 값의 변경이 필요하기 때문에 넣었습니다.
                d = 0;
            }
            if (!icheck(i)) {
                d++;
            }

            if (k == 5) {//네 방향 모두 청소가 이미 되어있거나 벽인 경우 후진하기 위한 함수
                if (od == 3) { //후진 하는 함수
                    ny = iny + by[od - 2];
                    nx = inx;
                    if (arr[ny][nx] == 0) { //만약에 벽인경우에는 통과 할 수 없기 때문에 검사합니다.
                        q.push({ny, nx});
                        d = od;
                    } else {  //이외의 경우의 수는 네 방향 모두 청소가 이미 되어있거나 벽이면서, 뒤쪽 방향이 벽이라 후진도 할 수 없는 경우라고 생각했습니다.
                        return;
                    }

                }
                if(od==1){
                    ny = iny + by[od + 2];
                    nx = inx;
                    if (arr[ny][nx] == 0) {
                        q.push({ny, nx});
                        d = od;
                    } else {
                        return;
                    }
                }
                if (od == 2) {
                    nx = inx + bx[od - 2];
                    ny = iny;
                    if (arr[ny][nx] == 0) {
                        q.push({ny, nx});
                        d = od;
                    } else {
                        return;
                    }
                }
                if (od == 0) {
                    nx = inx + bx[od + 2];
                    ny = iny;
                    if (arr[ny][nx] == 0) {
                        q.push({ny, nx});
                        d = od;
                    } else {
                        return;
                    }
                }
            }


        }


    }
}

    int main() {

        scanf("%d %d", &Y, &X);
        scanf("%d %d %d", &ry, &rx, &d);

        for (int i = 0; i < Y; i++) {
            for (int j = 0; j < X; j++) {
                scanf(" %d", &arr[i][j]);
            }
        }
        q.push({ry, rx});
        visited[ry][rx] = 1;


        bfs();


            printf("%d",cnt);
            return 0;


    }