반응형
1719번: 택배
첫째 줄에 두 수 n과 m이 빈 칸을 사이에 두고 순서대로 주어진다. n은 집하장의 개수로 200이하의 자연수, m은 집하장간 경로의 개수로 10000이하의 자연수이다. 이어서 한 줄에 하나씩 집하장간 경
www.acmicpc.net
알고리즘 종류
- 최단거리
- 다익스트라
사고 과정
- 다익스트라에서 다음 노드 정보를 우선순위 큐에 저장할 때, 가장 먼저 들린 집하장 노드도 넣어서 저장한다.
구현(C++)
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
struct Node{
int first = 0;
int cost = 987654321;
};
int n, m;
Node mat[201][201];
vector<vector<pair<int, int> > > bag; // from, to, cost
void dijkstra(int node){
priority_queue<pair<int, pair<int,int> > > pq;
// 초기화
pq.push({0, {node, 0}}); // cost, from, first
mat[node][node].cost = 0;
while(!pq.empty()){
int cost = -pq.top().first;
int from = pq.top().second.first;
int first = pq.top().second.second;
pq.pop();
for(int i=0; i<bag[from].size(); i++){
int next = bag[from][i].first;
int next_cost = cost + bag[from][i].second;
if(mat[node][next].cost > next_cost){
mat[node][next].cost = next_cost;
// 가장 먼저 거치는 노드가 없으면, 넣어준다. 있으면, 계속 유지해준다.
if(first == 0) mat[node][next].first = next;
else mat[node][next].first = first;
pq.push({-next_cost, {next, mat[node][next].first}});
}
}
}
}
void solution(){
// 각 노드에서 다익스트라 동작
for(int i=1; i<=n; i++)
dijkstra(i);
for(int i=1; i<=n; i++){
for(int j=1; j<=n; j++){
if(mat[i][j].first == 0) cout << "- ";
else cout << mat[i][j].first << " ";
}
cout << endl;
}
}
int main(void){
ios_base::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
cin >> n >> m;
bag.resize(n+1); // 간선 정보를 넣기 위해서 사이즈 조정
int a, b, c;
for(int i=0; i<m; i++){
cin >> a >> b >> c;
bag[a].push_back({b,c});
bag[b].push_back({a,c});
}
solution();
}
|
cs |
시행착오
반응형