백준 4485번 녹색 옷 입은 애가 젤다지?

문제 링크입니다: https://www.acmicpc.net/problem/4485

4485번: 녹색 옷 입은 애가 젤다지?

기존에 풀었던 다익스트라 문제들과는 달리 2차원 행렬을 입력받아 풀어야했던 문제였습니다.

전형적인 BFS 문제처럼 풀면 되고 여기에 가중치의 대소 관계만 고려한다면 쉽게 풀 수 있을 문제였습니다.

 

	#include <iostream>
	#include <vector>
	#include <algorithm>
	#include <functional>
	#include <queue>
	#include <cstring>
	using namespace std;
	const int INF = 987654321;
	const int MAX = 125 + 1;
	typedef struct
	{
	int y, x;
	}Dir;
	Dir moveDir[4] = { {1, 0}, {-1, 0}, {0, 1}, {0, -1} };
	int graph[MAX][MAX];
	int dist[MAX][MAX];
	bool visited[MAX][MAX];
	int main(void)
	{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int cnt = 1;
	while (1)
	{
	int N;
	cin >> N;
	if (N == 0)
	break;
	// memset이 INF에 대해서는 정상 작동 X
	for (int i = 0; i < N; i++)
	for (int j = 0; j < N; j++)
	dist[i][j] = INF;
	memset(visited, false, sizeof(visited));
	for (int i = 0; i < N; i++)
	for (int j = 0; j < N; j++)
	cin >> graph[i][j];
	priority_queue<pair<int, pair<int, int>>, vector<pair<int, pair<int, int>>>, greater<pair<int, pair<int, int>>>> pq;
	pq.push({ 0, {0, 0} });
	visited[0][0] = true;
	while (!pq.empty())
	{
	int y = pq.top().second.first;
	int x = pq.top().second.second;
	int cost = pq.top().first;
	pq.pop();
	for (int i = 0; i < 4; i++)
	{
	int nextY = y + moveDir[i].y;
	int nextX = x + moveDir[i].x;
	int nextCost = cost + graph[nextY][nextX];
	// 범위 내에 있고
	if(0 <= nextY && nextY < N && 0 <= nextX && nextX < N)
	// 아직 방문하지 않았으며, 누적합이 더 적을 때에만
	if (!visited[nextY][nextX] && dist[nextY][nextX] > nextCost)
	{
	visited[nextY][nextX] = true;
	dist[nextY][nextX] = nextCost;
	pq.push({ nextCost, {nextY, nextX} });
	}
	}
	}
	// 초기 우선순위 큐에 값을 넣을 때 graph[0][0]은 처리 안했으므로 같이 더해줘야한다.
	cout << "Problem " << cnt++ << ": " << graph[0][0] + dist[N - 1][N - 1] << "\n";
	}
	return 0;
	}

view raw .cpp hosted with ❤ by GitHub

 

개발환경:Visual Studio 2017

 

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~