| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 979468 | HappyCapybara | 사이버랜드 (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "cyberland.h"
#include<bits/stdc++.h>
#define int long long
using namespace std;
double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr){
vector<vector<pair<int,int>>> g(N);
for (int i=0; i<M; i++){
g[x[i]].push_back({y[i], c[i]});
g[y[i]].push_back({x[i], c[i]});
}
int dist = -1;
vector<bool> seen(N, false);
priority_queue<pair<int,int>> pq;
pq.push({0, 0});
while (!pq.empty()){
int d = -pq.top().first;
int cur = pq.top().second;
pq.pop();
if (cur == H){
dist = d;
break;
}
if (seen[cur]) continue;
seen[cur] = true;
for (pair<int,int> next : g[cur]) pq.push({-(d+next.second), next.first});
}
unordered_set<int> s;
seen.clear();
seen.resize(N, false);
queue<int> q;
q.push(0);
while (!q.empty()){
int cur = q.front();
q.pop();
if (cur == H) continue;
if (seen[cur]) continue;
seen[cur] = true;
if (arr[cur] == 0) s.insert(cur);
for (pair<int,int> next : g[cur]) q.push(next.first);
}
seen.clear();
seen.resize(N, false);
while (!pq.empty()) pq.pop();
pq.push({0, H});
while (!pq.empty()){
int d = -pq.top().first;
int cur = pq.top().second;
pq.pop();
if (s.find(cur) != s.end()){
if (d < dist) return (double) d;
else break;
}
if (seen[cur]) continue;
seen[cur] = true;
for (pair<int,int> next : g[cur]) pq.push({-(d+next.second), next.first});
}
return (double) dist;
}