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

안녕하세요 큰돌님.

문제 해답 보기 전에 영상 멈추고 풀기 시작했다가 1시간이나 까먹었네요.... 이래도 성장 할 수 있을지 고민입니다.

개인적인 고민은 뒤로 두고 다음의 코드로 풀긴 했는데 이게 맞는 코드인지 봐주실수 있나요?

항상 질문 답변 해주셔서 감사합니다.

#include <bits/stdc++.h>

using namespace std;

int N, M, startY, startX, destY, destX, adj[104][104];
int dy[] = {1, 0, -1, 0};
int dx[] = {0, 1, 0, -1};
map<pair<int, int>, int> visited;

void bfs(int y, int x)
{
    pair<int, int> pos = {y, x};
    visited[pos] += 1;

    queue<pair<int, int>> q;
    q.push(pos);

    while (q.size())
    {
        auto loc = q.front();
        q.pop();

        for (int i = 0; i < 4; ++i)
        {
            int ny = loc.first + dy[i];
            int nx = loc.second + dx[i];
            pair<int, int> prevOffset = {loc.first, loc.second};
            pair<int, int> nextOffset = {ny, nx};
            if (ny < 0 || nx < 0 || ny >= M || nx >= N)
                continue;
            if (visited[nextOffset])
                continue;
            if (adj[ny][nx] == 0)
                continue;

            visited[nextOffset] = visited[prevOffset] + 1;
            q.push(nextOffset);
            if (ny == destY && nx == destX)
                q = queue<pair<int, int>>();
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> N >> M;
    cin >> startY >> startX;
    cin >> destY >> destX;

    for (int i = 0; i < N; ++i)
        for (int j = 0; j < M; ++j)
            cin >> adj[i][j];

    bfs(startY, startX);

    cout << visited[make_pair(destY, destX)] << "\n";

    return 0;
}