This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define FOR(a, b, c) for(int a = b; a <= c; ++a)
#define pb push_back
#define int long long
typedef pair<int, int> ii;
const int N = 1e5 + 10;
const int oo = 1e18;
int n, m, k;
int d[(1 << 6)][N];
vector<ii> vi[N];
signed main()
{
scanf("%lld%lld%lld", &n, &k, &m);
FOR(i, 1, n) FOR(j, 0, (1 << k)) d[j][i] = oo;
FOR(i, 1, k) {
int x; scanf("%lld", &x);
d[(1 << (i - 1))][x] = 0;
}
FOR(i, 1, m) {
int u, v, c; scanf("%lld%lld%lld", &u, &v, &c);
vi[u].pb(ii(v, c)); vi[v].pb(ii(u, c));
}
FOR(mask, 1, (1 << k) - 1) {
FOR(j, 1, n) for(int submask = mask; submask; submask = (submask - 1) & mask)
d[mask][j] = min(d[mask][j], d[mask ^ submask][j] + d[submask][j]);
priority_queue<ii, vector<ii>, greater<ii> > pq;
FOR(i, 1, n) pq.push(ii(d[mask][i], i));
while(!pq.empty()) {
int u = pq.top().se, du = pq.top().fi; pq.pop();
if(du != d[mask][u]) continue;
for(auto v: vi[u]) if(d[mask][v.fi] > du + v.se) {
d[mask][v.fi] = du + v.se;
pq.push(ii(du + v.se, v.fi));
}
}
}
int ans = oo;
FOR(i, 1, n) ans = min(ans, d[(1 << k) - 1][i]);
printf("%lld", ans);
}
Compilation message (stderr)
cities.cpp: In function 'int main()':
cities.cpp:21:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%lld%lld%lld", &n, &k, &m);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
cities.cpp:24:21: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
int x; scanf("%lld", &x);
~~~~~^~~~~~~~~~~~
cities.cpp:28:27: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
int u, v, c; scanf("%lld%lld%lld", &u, &v, &c);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |