[백준] 14284번. 간선 이어가기 2

leeeha·2024년 6월 18일
0

백준

목록 보기
171/186
post-thumbnail
post-custom-banner

문제

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


풀이

완탐 (메모리 초과)

#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 = 5002;
vector<pii> graph[MAX];
int n, m, s, e;
int ans = 1e9;

void input() {
	cin >> n >> m;  

	for(int i = 0; i < m; i++){
		int a, b, c;
		cin >> a >> b >> c; 
		graph[a].push_back({b, c});
	}

	cin >> s >> e;
}

void dfs(int now, int result){
	if(now == e){
		ans = min(ans, result);
		return; 
	}
	
	for(auto edge: graph[now]){
		int adj = edge.first; 
		int cost = edge.second; 
		dfs(adj, result + cost); 
	}
}

void solution() {
	dfs(s, 0);
	cout << ans; 
}

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

	input(); 
	solution();

	return 0;
}

다익스트라

한 노드에서 다른 모든 노드까지의 최단 경로를 구할 때 사용되는 다익스트라 알고리즘을 이용하면, s 노드에서 e 노드까지의 최단 거리도 쉽게 구할 수 있다.

#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 = 5001;
const int INF = 1e9; 
vector<pii> graph[MAX];
int dist[MAX]; 
int n, m, s, e;

void input() {
	cin >> n >> m;  

	for(int i = 0; i < m; i++){
		int a, b, c;
		cin >> a >> b >> c; 

		// 양방향 연결 
		graph[a].push_back({b, c});
		graph[b].push_back({a, c});
	}

	cin >> s >> e;

	// 최단 거리 테이블 초기화 
	fill_n(dist, n + 1, INF); 
}

void dijkstra() {
	// 시작 노드로부터의 거리를 기준으로 한 최소 힙  
	priority_queue<pii, vector<pii>, greater<pii>> pq;
	pq.push({0, s}); // 거리, 노드 번호 
	dist[s] = 0;

	while(!pq.empty()){
		int curDist = pq.top().first;
		int now = pq.top().second;
		pq.pop();

		// 한번 처리된 노드는 무시 (이미 경로 확정)
		if(dist[now] < curDist) continue;

		// 현재 노드와 인접한 노드 탐색 
		for(auto edge: graph[now]){
			int next = edge.first; 
			int cost = edge.second; 
			int newCost = cost + curDist; 

			if(dist[next] > newCost){
				dist[next] = newCost; 
				pq.push({newCost, next}); 
			}
		}
	}
}

void solution() {
	dijkstra(); 
	cout << dist[e]; 
}

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

	input(); 
	solution();

	return 0;
}
profile
습관이 될 때까지 📝
post-custom-banner

0개의 댓글