이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#pragma once
#include <vector>
#include <iostream>
#include <set>
using namespace std;
const int N1=1e5+100;
#define ll long long
vector<pair<int,int>> ma[N1];
ll dp[N1],dip[N1][2];
ll ans[N1][2];
bool vis[N1],done_for[N1];
void dyk()
{
set<pair<ll,ll>> dist;
dist.insert({0,0});
while(dist.size()>0)
{
auto f=*begin(dist);
dist.erase(begin(dist));
for(auto [w,y]:ma[f.second])
{
if(!done_for[y] and dp[y]>(f.first+w))
{
dist.erase({dp[y],y});
dp[y]=f.first+w;
dist.insert({dp[y],y});
}
}
}
}
int travel_plan(int n, int m, int r[][2], int l[], int k, int p[])
{
for(int i=0;i<n;i++)
dp[i]=dip[i][0]=dip[i][1]=1e11;
for(int i=0;i<m;i++)
{
ma[r[i][0]].push_back({l[i],r[i][1]});
ma[r[i][1]].push_back({l[i],r[i][0]});
}
for(int i=0;i<k;i++)
{
for(auto [w,y]:ma[p[i]])
{
if(dip[y][0]>w)
{
dip[y][1]=dip[y][0];
dip[y][0]=w;
}
else if(dip[y][1]>w)
dip[y][1]=w;
}
done_for[p[i]]=1;
}
if(done_for[0])
return 0;
dyk();
ll ans=1e11;
for(int i=0;i<n;i++)
{
ans=min(ans,dip[i][1]+dp[i]);
// ans[i][0]=dip[i][1]+dp[i];
}
// cout<<dip[0][0]<<' '<<dip[0][1]<<endl;
return ans;
}
컴파일 시 표준 에러 (stderr) 메시지
crocodile.cpp:1:9: warning: #pragma once in main file
1 | #pragma once
| ^~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |