인프런 커뮤니티 질문&답변

좋은 수업 잘 듣고 있습니다.

먼저 제 코드를 짜고 선생님 수업에 대한 내용을 확인하는 편인데, 일단 제 로직과 선생님의 로직에 약간의 차이는 있는 소스코드입니다.

저같은 경우에는, while 루프를 돌 때, 먼저 cnt라는 것이 현재 들어이는 Q의 size를 넣어줍니다. 그렇게 되면 level 1에서 찾아야하는 토마토 개수가 들어갈 것이고 해당하는 만큼 for문을 돌게 되면서 상하좌우를 서치하고 다시 Q에 넣어줍니다. Q에 넣더라도 처음의 Q size만큼 for문을 돌기 때문에 level 2로 넘어가지 않고 for문이 끝나고 다시 while의 처음으로 들어오게되는 정도의 차이정도가 있습니다. 39행부터 56행까지의 내용입니다.

이 로직이 틀린걸까요? 아니면 제가 틀린걸까요? test를 돌려보면 1, 2, 3번 파일은 맞추나 4, 5번 파일이 wrong_answer가 떠서 60점이 나옵니다.

소스코드는 아래와 같습니다.

감사합니다.

#include<cstdio>

#include<queue>

#include<malloc.h>

using namespace std;

struct Loc{

  int x;

  int y;

  Loc(int a, int b){

    x = a;

    y = b;

  }

};

int main(){

  int n, m;

  int **arr;

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

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

  scanf("%d %d", &n, &m);

  arr = (int **)malloc(sizeof(int) * m);

  for(int i=0; i<m; i++){

    arr[i] = (int *)malloc(sizeof(int) * n);

  }

  for(int i=0; i<m; i++){

    for(int j=0; j<n; j++){

      scanf("%d", &arr[i][j]);

    }

  }

  queue<Loc> Q;

  for(int i=0; i<m; i++){

    for(int j=0; j<n; j++){

      if(arr[i][j] == 1){

        Q.push(Loc(j, i));

      }

    }

  }

  int day = 0;

  while(!Q.empty()){

    int cnt = Q.size();

    for(int i=0; i<cnt; i++){

      Loc tmp = Q.front();

      Q.pop();

      for(int j=0; j<4; j++){

        int xx = tmp.x + dx[j];

        int yy = tmp.y + dy[j];

        if(xx >= 0 && xx < n && yy >= 0 && yy < m){

          if(arr[yy][xx] == 0){

            arr[yy][xx] = 1;

            Q.push(Loc(xx, yy));

          }

        }

      }

    }

    day++;

  }

  printf("%d", day-1);

  for(int i=0; i<m; i++) free(arr[i]);

  free(arr);

}