작성한 질문수 6

0

안녕하세요 ㅎㅎ

사실 그부분만 빼면 괜찮은 코드입니다.

저도 처음에는 그 부분 살려가면서 고치려고 했는데 왜인지 모르게 메모리초과나 틀렸습니다가 뜨더군요.

그럴 때 저는 보통 코드 다듬기를 합니다.

일단.. 저부분은 일단은 비효율적이기 때문에 다듬고.. 다른 부분도 최대한 수강생님 코드 살려가면서 다듬어봤는데요.

#include <bits/stdc++.h>
using namespace std;

int R, C;
int fire_visited[1004][1004];
int jihoon_visited[1004][1004];
int dy[4] = {-1, 0, 1, 0}, dx[4] = {0, 1, 0, -1};
char arr[1004][1004];
pair<int, int> jihoon;
vector<pair<int, int>> fire_v;

void bfs_fire() {
    queue<pair<int, int>> q;
    for (auto &f : fire_v) {
        q.push(f);
        fire_visited[f.first][f.second] = 1;
    }

    while (!q.empty()) {
        auto [y, x] = q.front();
        q.pop();
        for (int i = 0; i < 4; i++) {
            int ny = y + dy[i], nx = x + dx[i];
            if (ny >= 0 && nx >= 0 && ny < R && nx < C && arr[ny][nx] != '#' && (!fire_visited[ny][nx] || fire_visited[ny][nx] > fire_visited[y][x] + 1)) {
                fire_visited[ny][nx] = fire_visited[y][x] + 1;
                q.push({ny, nx});
            }
        }
    }
}

int bfs_jihoon() {
    queue<pair<int, int>> q;
    q.push(jihoon);
    jihoon_visited[jihoon.first][jihoon.second] = 1;

    while (!q.empty()) {
        auto [y, x] = q.front();
        q.pop();
        for (int i = 0; i < 4; i++) {
            int ny = y + dy[i], nx = x + dx[i];
            if (ny < 0 || nx < 0 || ny >= R || nx >= C) return jihoon_visited[y][x]; 
            if (!jihoon_visited[ny][nx] && arr[ny][nx] != '#' && (fire_visited[ny][nx] == 0 || jihoon_visited[y][x] + 1 < fire_visited[ny][nx])) {
                jihoon_visited[ny][nx] = jihoon_visited[y][x] + 1;
                q.push({ny, nx});
            }
        }
    }
    return 0;  
}

int main() {
    cin >> R >> C;
    memset(fire_visited, 0, sizeof(fire_visited));

    for (int i = 0; i < R; i++) {
        for (int j = 0; j < C; j++) {
            cin >> arr[i][j];
            if (arr[i][j] == 'J') jihoon = {i, j};
            else if (arr[i][j] == 'F') {
                fire_v.push_back({i, j});
                fire_visited[i][j] = 1;
            }
        }
    }

    bfs_fire();  
    int answer = bfs_jihoon(); 

    if (answer) cout << answer;
    else cout << "IMPOSSIBLE";

    return 0;
}

이렇게 한번 해보시겠어요?

전체적인 로직은 정말 잘 짜셨습니다. ㅎㅎ

틀린부분을 명확하게 알려주고 싶은데 마음대로 되지 않네요.

그래도 최대한 다듬어 봤습니다. ㅎㅎ

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.