#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mp make_pair
#define pb push_back
#define x first
#define y second
#define pii pair<int, int>
#define p3i pair<pii, int>
#define pll pair<ll, ll>
#define p3l pair<pll, ll>
#define lseg L, (L+R)/2, N*2+1
#define rseg (L+R)/2+1, R, N*2+2
#define ub upper_bound
#define lb lower_bound
#define pq priority_queue
#define MN 1000000007
#define fox(k, x) for (int k=0; k<x; ++k)
#define fox1(k, x) for (int k=1; k<=x; ++k)
#define foxr(k, x) for (int k=x-1; k>=0; --k)
#define fox1r(k, x) for (int k=x; k>0; --k)
#define ms multiset
#define flood(x) memset(x, 0x3f3f3f3f, sizeof x)
#define drain(x) memset(x, 0, sizeof x)
#define rng() (rand() >> 3)*rand()
int n, m, k, a, b, c, N, B;
ll dis[32][100005], D;
bool u[32][100005];
const ll t=(1LL << 60);
vector<pii> v[100005];
pq<pair<ll, int> > q;
int main(){
scanf("%i%i%i", &n, &k, &m);
fox(l, (1 << k)){
fox1(l2, n) dis[l][l2]=1152921504606846976LL;
}
fox(l, k){
scanf("%i", &a);
dis[1 << l][a]=0;
q.push(mp(0, (1 << l) + (a << 5)));
}
fox(l, m){
scanf("%i%i%i", &a, &b, &c);
v[a].pb(mp(b, c));
v[b].pb(mp(a, c));
}
while(!q.empty()){
N=(q.top().y >> 5); D=-q.top().x; B=(q.top().y&31); q.pop();
if (u[B][N]) continue;
u[B][N]=1;
//cout << N << ' ' << B << endl;
if (B==(1 << k)-1){
cout << D << endl;
return 0;
}
fox(l, (1 << k)){
if (dis[B|l][N]>D+dis[l][N]){
dis[B|l][N]=D+dis[l][N];
q.push(mp(-D-dis[l][N], (B|l)+(N << 5)));
}
}
fox(l, v[N].size()){
if (dis[B][v[N][l].x]>D+v[N][l].y){
dis[B][v[N][l].x]=D+v[N][l].y;
q.push(mp(-D-v[N][l].y, B+(v[N][l].x << 5)));
}
}
}
return 0;
}
Compilation message
cities.cpp: In function 'int main()':
cities.cpp:18:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
#define fox(k, x) for (int k=0; k<x; ++k)
^
cities.cpp:63:9: note: in expansion of macro 'fox'
fox(l, v[N].size()){
^
cities.cpp:34:32: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%i%i%i", &n, &k, &m);
^
cities.cpp:39:24: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%i", &a);
^
cities.cpp:44:36: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%i%i%i", &a, &b, &c);
^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
32484 KB |
Output is correct |
2 |
Correct |
0 ms |
32484 KB |
Output is correct |
3 |
Correct |
0 ms |
32484 KB |
Output is correct |
4 |
Correct |
0 ms |
32484 KB |
Output is correct |
5 |
Correct |
0 ms |
32484 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
729 ms |
63368 KB |
Output is correct |
2 |
Correct |
656 ms |
51012 KB |
Output is correct |
3 |
Correct |
169 ms |
37228 KB |
Output is correct |
4 |
Correct |
106 ms |
37700 KB |
Output is correct |
5 |
Correct |
259 ms |
44932 KB |
Output is correct |
6 |
Correct |
103 ms |
37292 KB |
Output is correct |
7 |
Correct |
0 ms |
32752 KB |
Output is correct |
8 |
Correct |
3 ms |
32616 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
33400 KB |
Output is correct |
2 |
Correct |
6 ms |
33008 KB |
Output is correct |
3 |
Correct |
3 ms |
32756 KB |
Output is correct |
4 |
Correct |
3 ms |
33008 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2893 ms |
87940 KB |
Output is correct |
2 |
Correct |
2336 ms |
87860 KB |
Output is correct |
3 |
Correct |
1586 ms |
60268 KB |
Output is correct |
4 |
Correct |
1386 ms |
87448 KB |
Output is correct |
5 |
Correct |
216 ms |
50028 KB |
Output is correct |
6 |
Correct |
109 ms |
39416 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
4000 ms |
235396 KB |
Execution timed out |
2 |
Halted |
0 ms |
0 KB |
- |