| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 815561 | OAleksa | 악어의 지하 도시 (IOI11_crocodile) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
//#include "factories.h"
//#include "crocodile.h"
#define f first
#define s second
using namespace std;
const int maxn = 1e5 + 69;
set<pair<int, int>> g[maxn];
vector<int> cnt(maxn), par(maxn, -1), d(maxn, 1e9);
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
for(int i = 0;i < M;i++) {
g[R[i][0]].insert({L[i], R[i][1]});
g[R[i][1]].insert({L[i], R[i][0]});
}
for(int i = 0;i < K;i++)
cnt[P[i]] = 1;
for(int i = 1;i < N;i++) {
if(cnt[i])
continue;
if(g[i].size() <= 2 && !cnt[i]) {
for(auto x : g[i])
g[x.s].erase({x.f, i});
g[i].clear();
}
}
// for(int i = 0;i < N;i++) {
// cout << i << endl;
// for(auto x : g[i]) {
// cout << x.f << " " << x.s << endl;
// }
// cout << endl;
// }
priority_queue<pair<int, int>> pq;
d[0] = 0;
pq.push({0, 0});
while(!pq.empty()) {
auto u = pq.top();
pq.pop();
auto x = *g[u.s].begin();
if(x.s == par[u.s])
g[u.s].erase(++g[u.s].begin());
else
g[u.s].erase(g[u.s].begin());
for(auto v : g[u.s]) {
if(d[v.s] <= d[u.s] + v.f)
continue;
d[v.s] = d[u.s] + v.f;
pq.push({-d[v.s], v.s});
}
}
int ans = 1e9;
for(int i = 1;i < N;i++) {
if(cnt[i])
ans = min(ans, d[i]);
}
return ans;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int tt = 1;
//cin >> tt;
while(tt--) {
int n, m;
cin >> n >> m;
int r[m][2];
for(int i = 0;i < m;i++)
cin >> r[i][0] >> r[i][1];
int l[m];
for(int i = 0;i < m;i++)
cin >> l[i];
int k;
cin >> k;
int p[k];
for(int i = 0;i < k;i++)
cin >> p[i];
cout << travel_plan(n, m, r, l, k, p);
}
return 0;
}