-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTRAFFICN.cpp
More file actions
104 lines (96 loc) · 1.96 KB
/
TRAFFICN.cpp
File metadata and controls
104 lines (96 loc) · 1.96 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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;
typedef pair<int, lli> pii;
typedef pair<int, lli> pii2;
#define INF 1000010000
vector<pii> graph[10010];
vector<pii> graphrev[10010];
lli d[10010];
lli dr[10010];
bool done[10010];
class comp {
public:
bool operator() (const pii& a, const pii& b) {
return a.second > b.second;
}
};
int main() {
int t1;
scanf("%d",&t1);
while(t1--) {
int n,m,e,s,t;
scanf("%d%d%d%d%d",&n,&m,&e,&s,&t);
for (int i = 0; i <= 10010; ++i)
{
graph[i].clear();
graphrev[i].clear();
}
for(int i = 0; i < m; i++) {
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
graph[u].push_back(make_pair(v,w));
graphrev[v].push_back(make_pair(u,w));
}
//For First Graph
priority_queue<pii2, vector<pii2>, comp> pq;
pq.push(make_pair(s,0));
memset(done, false, sizeof(done));
for (int i = 0; i <= n; ++i)
d[i] = INF;
d[s] = 0;
while(!pq.empty()) {
int u = pq.top().first;
pq.pop();
for (int i = 0; i < graph[u].size(); ++i)
{
int v = graph[u][i].first;
lli w = graph[u][i].second;
if(!done[v] && d[v] > (d[u] + w)) {
d[v] = d[u] + w;
pq.push(make_pair(v,d[v]));
}
}
done[u] = true;
}
//For Reverse Graph
pq.push(make_pair(t,0));
memset(done, false, sizeof(int));
for (int i = 0; i <= n; ++i)
dr[i] = INF;
dr[t] = 0;
while(!pq.empty()) {
int u = pq.top().first;
pq.pop();
for (int i = 0; i < graphrev[u].size(); ++i)
{
int v = graphrev[u][i].first;
lli w = graphrev[u][i].second;
if(!done[v] && dr[v] > (dr[u] + w)) {
dr[v] = dr[u] + w;
pq.push(make_pair(v,dr[v]));
}
}
done[u] = true;
}
//Final Answer
lli ans = d[t];
for (int i = 0; i < e; ++i)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
lli l1 = (d[u]+dr[v]+w);
lli l2 = (d[v]+dr[u]+w);
if(l1 < ans) {
ans = l1;
}
if(l2 < ans)
ans = l2;
}
if(ans == INF)
printf("-1\n");
else
printf("%lld\n", ans);
}
return 0;
}