| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 881646 | tsumondai | 사이버랜드 (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define foru(i, l, r) for(int i = l; i <= r; i++)
#define ford(i, r, l) for(int i = r; i >= l; i--)
#define __TIME (1.0 * clock() / CLOCKS_PER_SEC)
using vi = vector<int>;
using ar = pair<double,int>;
typedef pair<int, int> ii;
typedef pair<ii, int> iii;
typedef pair<ii, ii> iiii;
typedef pair<double, int> ar;
const int N = 1e5 + 5;
const int oo = 1e9, mod = 1e9 + 7;
int n, m, h;
double ans = 0;
bool vis[N];
double dis[N], dis2[N];
vector<ii> adj[N];
priority_queue<ar, vector<ar>, greater<ar>> pq;
void dijkstra(){
fill(vis,vis+n+1,0);
for(int i = 0; i < n; i++)
if(dis[i]<oo) pq.push({dis[i],i});
while(!pq.empty()){
auto tmp = pq.top(); pq.pop();
double D=tmp.fi; int a=tmp.se;
if(vis[a] || a==h) continue; vis[a]=1;
for(auto tmp : adj[a]) {
int b=tmp.fi, w=tmp.se;
if(dis[b]>dis[a]+w)
dis[b]=dis[a]+w, pq.push({dis[b],b});
}
}
ans = min(ans, dis[h]);
}
double solve(int N, int M, int K, int H, vi x, vi y, vi c, vi A) {
n = N, m = M, h = H, K = min(K, 67ll);
fill(dis, dis + n, oo); ans = oo;
for (int i = 0; i < n; i++) adj[i].clear();
for (int i = 0; i < m; i++)
adj[x[i]].push_back({y[i], c[i]}),
adj[y[i]].push_back({x[i], c[i]});
dis[0] = 0; dijkstra(); if (dis[h] >= oo) return -1;
for (int i = 0; i < n; i++)
if (dis[i] != oo) dis[i] = A[i] && i ? oo : 0;
for (int i = 0; i <= K; i++) {
dijkstra(); fill(dis2, dis2 + n, oo);
for (int j = 0; j < n; j++)
if (A[j] > 1) for (auto tmp : adj[j]) {
int u=tmp.fi, w=tmp.se;
dis2[u] = min(dis2[u], dis[j] / 2 + w);
}
for (int j = 0; j < n; j++) dis[j] = dis2[j];
}
return ans;
}