[백준] 14940번. 쉬운 최단거리

leeeha·2024년 8월 5일
0

백준

목록 보기
182/186
post-thumbnail

문제

https://www.acmicpc.net/problem/14940


풀이

처음에 이 문제를 풀 때 목표 지점인 2번을 글자 그대로 '도착점'으로 인식하고 푸니까 풀이가 꼬였다.

다른 모든 지점부터 2번까지의 거리를 구하려면 출발 노드에 따라 BFS를 여러 번 돌려야 하는데

2번을 출발점이라 생각하고 다른 모든 지점까지의 거리를 구하려면, BFS를 한번만 돌리면 된다!

그리고 2번부터 각 노드까지의 거리를 dist라는 별도의 2차원 배열에 저장해주면 된다.

한 노드를 여러 번 방문할 수 있기 때문에 방문 여부를 체크하는 배열은 따로 필요하지 않다.

이번 문제는 목표 지점인 2번을 도착점이 아니라 '시작점'으로 인식하는 것이 중요한 문제였다. 생각의 전환이 중요하다!

#include <iostream>
#include <string> 
#include <cstring>
#include <sstream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;

const int MAX = 1001;
int N, M;

int graph[MAX][MAX]; // 0: 벽, 1: 빈칸, 2: 목표 지점
int dist[MAX][MAX]; // 2부터 다른 지점까지의 거리 
queue<pii> q;

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

void input(){
	cin >> N >> M;

	for(int i = 0; i < N; i++){
		for(int j = 0; j < M; j++){
			dist[i][j] = -1;
		}
	}
	
	for(int i = 0; i < N; i++){
		for(int j = 0; j < M; j++){
			cin >> graph[i][j];

			// 목표 지점 위치 저장 
			if(graph[i][j] == 2){
				q.push({i, j});
				dist[i][j] = 0;
			}
		}
	}
}

void bfs(int x, int y){
	// 2부터 다른 모든 지점까지의 최단 거리 계산 
	// 도달할 수 없는 경우 -1 그대로 출력 
	while(!q.empty()){
		auto now = q.front();
		q.pop();
		int row = now.first;
		int col = now.second;
		
		for(int i = 0; i < 4; i++){
			int nx = row + dx[i];
			int ny = col + dy[i];
			
			if(nx < 0 || nx >= N || ny < 0 || ny >= M) continue;
			if(dist[nx][ny] != -1) continue;
			if(graph[nx][ny] == 0) continue;

			q.push({nx, ny});
			dist[nx][ny] = dist[row][col] + 1;
		}
	}
}

void solution(){
	for(int i = 0; i < N; i++){
		for(int j = 0; j < M; j++){
			if(graph[i][j] == 0){
				cout << "0 ";
				continue;
			}

			bfs(i, j);
			cout << dist[i][j] << " ";
		}
		cout << "\n";
	}
}

int main()
{ 
	ios::sync_with_stdio(0);
	cin.tie(0);

	input();
	solution();

	return 0;
}
profile
습관이 될 때까지 📝

0개의 댓글