-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathP1339.cc
More file actions
70 lines (61 loc) · 1.51 KB
/
P1339.cc
File metadata and controls
70 lines (61 loc) · 1.51 KB
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
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <queue>
using std::vector;
using std::priority_queue;
#define N 2500
struct AdjNode {
int adjv;
int weight;
AdjNode(int adjv, int weight) : adjv(adjv), weight(weight) {}
};
vector<AdjNode> graph[N];
int dist[N];
bool collected[N];
struct PNode{
int v;
int dist;
PNode(int v, int dist) : v(v), dist(dist) {}
};
struct cmp {
bool operator()(PNode& a, PNode& b) {
return a.dist > b.dist;
}
};
priority_queue<PNode, vector<PNode>, cmp> q;
inline int dijkstra(int s, int t) {
memset(dist, 0xff, sizeof(dist)); //-1
memset(collected, 0, sizeof(collected)); //false
dist[s] = 0;
q.push(PNode(s, 0));
while (!q.empty()) {
int v = q.top().v;
q.pop();
if (collected[v]) continue;
collected[v] = true;
if (v == t) break;
for (auto it = graph[v].begin(); it != graph[v].end(); it++) {
int w = it->adjv;
if (dist[w] == -1 || dist[v] + it->weight < dist[w]) {
dist[w] = dist[v] + it->weight;
q.push(PNode(w, dist[w]));
}
}
}
return dist[t];
}
int main() {
int n, m, s, t, u, v, w;
scanf("%d %d %d %d", &n, &m, &s, &t);
s--; t--;
for (int i = 0; i < m; i++) {
scanf("%d %d %d", &u, &v, &w);
u--; v--;
graph[v].push_back(AdjNode(u, w));
graph[u].push_back(AdjNode(v, w));
}
printf("%d", dijkstra(s, t));
exit(0);
}