#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<ll> vll;
const ll mod=1e9+7;
const ll maxn=1e5+5;
const ll INF=1e18;
#define ok ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define fi first
#define se second
#define pb push_back
#define ub upper_bound
#define lb lower_bound
#define endl '\n'
int n, k, m, arr[5], u, v, w;
ll dis[maxn][5], ans=INF;
vector<pii> adj[maxn];
int main() {
ok
cin >> n >> k >> m;
for(int i=0; i<k; i++) cin >> arr[i];
for(int i=0; i<m; i++) {
cin >> u >> v >> w;
adj[u].pb({v, w});
adj[v].pb({u, w});
}
for(int i=0; i<k; i++) {
for(int j=1; j<=n; j++) dis[j][i]=INF;
dis[arr[i]][i]=0;
priority_queue<pll, vector<pll>, greater<pll>> pq;
pq.push({0, arr[i]});
while(!pq.empty()) {
pll fr=pq.top();
pq.pop();
if(fr.fi>dis[fr.se][i]) continue;
for(auto it : adj[fr.se]) {
if(dis[it.fi][i]>dis[fr.se][i]+it.se) {
dis[it.fi][i]=dis[fr.se][i]+it.se;
pq.push({dis[it.fi][i], it.fi});
}
}
}
}
for(int i=1; i<=n; i++) {
ll cur=0;
for(int j=0; j<k; j++) cur+=dis[i][j];
ans=min(ans, cur);
}
cout << ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
2644 KB |
Output is correct |
2 |
Correct |
2 ms |
2644 KB |
Output is correct |
3 |
Correct |
2 ms |
2644 KB |
Output is correct |
4 |
Incorrect |
2 ms |
2684 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
227 ms |
18820 KB |
Output is correct |
2 |
Correct |
221 ms |
19936 KB |
Output is correct |
3 |
Correct |
88 ms |
11704 KB |
Output is correct |
4 |
Correct |
54 ms |
10856 KB |
Output is correct |
5 |
Correct |
207 ms |
18852 KB |
Output is correct |
6 |
Correct |
53 ms |
10792 KB |
Output is correct |
7 |
Correct |
3 ms |
2772 KB |
Output is correct |
8 |
Correct |
3 ms |
2804 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
2824 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
288 ms |
18912 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
327 ms |
18920 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |