| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 751817 | 1075508020060209tc | Cyberland (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
int n;int m;int KK;int HH;
ll ar[200005];
ll xar[200005];ll yar[200005];ll car[200005];
vector<pair<int,ll>>e[200005];
void inp(){
// cin>>n>>m>>KK>>HH;
HH++;
for(int i=1;i<=n;i++){
// cin>>ar[i];
e[i].clear();
}
for(int i=1;i<=m;i++){
//cin>>xar[i]>>yar[i]>>car[i];
xar[i]++;yar[i]++;
e[xar[i]].push_back({yar[i],car[i]});
e[yar[i]].push_back({xar[i],car[i]});
}
}
int uf[200005];
int fin(int x){
if(uf[x]==x){return x;}
uf[x]=fin(uf[x]);
return uf[x];
}
void mrg(int a,int b){
int pa=fin(a);int pb=fin(b);
if(pa==pb){return;}
uf[pa]=pb;
}
double dis[110][100005][2];
int vis[110][100005][2];
double slv(){
for(int i=1;i<=n;i++){
uf[i]=i;
}
for(int i=1;i<=m;i++){
if(xar[i]==HH||yar[i]==HH){continue;}
mrg(xar[i],yar[i]);
}
/*
*/
priority_queue<pair<double,pair<int,int>>,vector<pair<double,pair<int,int>>>,greater<pair<double,pair<int,int>>>>pq;
KK=min(KK,70);
for(int k=0;k<=KK;k++){
for(int i=1;i<=n;i++){
dis[k][i][0]=1e16;
vis[k][i][0]=0;
dis[k][i][1]=1e16;
vis[k][i][1]=0;
}
}
dis[0][1]=0;
pq.push({0,{0,1}});
for(int i=2;i<=n;i++){
if(ar[i]==0&&fin(i)==fin(1)){
pq.push({0,{0,i}});
dis[0][i][0]=0;
pq.push({0,{1,i}});
dis[0][i][1]=0;
}
}
for(int i=1;i<=m;i++){
mrg(xar[i],yar[i]);
}
if(fin(1)!=fin(HH)){
return -1;
}
while(pq.size()){
int nw=pq.top().second.second;
int nwv=pq.top().second.first;
pq.pop();
if(vis[0][nw][nwv]){continue;}
vis[0][nw][nwv]=1;
for(int i=0;i<e[nw].size();i++){
int v=e[nw][i];
double w=e[nw][i].second;
if(!vis[0][v][0]){
if(dis[0][v][0]>dis[0][nw][nwv]+w){
dis[0][v][0]=dis[0][nw][nwv]+w;
pq.push({dis[0][v][0],{0,v}});
}
}
}
}
double ans=dis[0][HH];
for(int i=1;i<=KK;i++){
if(!vis[i][HH]){continue;}
ans=min(ans,dis[i][HH]);
}
/*
for(int i=0;i<=3;i++){
for(int j=1;j<=n;j++){
cout<<dis[i][j]<<" ";
}cout<<endl;
}
*/
return ans;
}
//#define int int
double solve(int N, int M, int K, int H, std::vector<int> x, std::vector<int> y, std::vector<int> c, std::vector<int> arr){
cin>>N>>M>>K>>H;
for(int i=1;i<=N;i++){
int v;
cin>>v;
arr.push_back(v);
}
// cout<<M<<" ";
for(int i=1;i<=M;i++){
int a;int b;int cc;
cin>>a>>b>>cc;
x.push_back(a);y.push_back(b);c.push_back(cc);
// cout<<i<<" "<<M<<endl;
}
n=N;m=M;KK=K;HH=H;
for(int i=0;i<x.size();i++){
xar[i+1]=x[i];
yar[i+1]=y[i];
car[i+1]=c[i];
}
for(int i=0;i<arr.size();i++){
ar[i+1]=arr[i];
}
// cout<<"hehe";
inp();
return slv();
}
signed main()
{
cout<<fixed<<setprecision(7)<<solve(0,0,0,0,{},{},{},{});
// inp();
// cout<<slv();
return 0;
}