이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
vector<pair<int,long long> > A[100005];
long long Dist[2][100005];
long long Dp[3][100005];
priority_queue<pair<long long,pair<int,int> > > Q;
void getDists(int start1, int start2, int n)
{
for(int i=1; i<=n; i++)
Dist[0][i]=Dist[1][i]=1e18;
Dist[0][start1]=0;
Dist[1][start2]=0;
Q.push({0,{0,start1}});
Q.push({0,{1,start2}});
while(!Q.empty())
{
int wh=Q.top().second.first;
int nod=Q.top().second.second;
long long dist=-Q.top().first;
Q.pop();
if(dist!=Dist[wh][nod])
continue;
for(auto edge:A[nod])
{
if(Dist[wh][edge.first]>dist+edge.second)
{
Dist[wh][edge.first]=dist+edge.second;
Q.push({-Dist[wh][edge.first],{wh,edge.first}});
}
}
}
}
void better(int wh, int nod, long long cost)
{
if(Dp[wh][nod]>cost)
{
Dp[wh][nod]=cost;
Q.push({-cost,{wh,nod}});
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n,m,s,t,u,v;
cin>>n>>m>>s>>t>>u>>v;
for(int i=1; i<=m; i++)
{
int x,y,c;
cin>>x>>y>>c;
A[x].push_back({y,c});
A[y].push_back({x,c});
}
getDists(s,t,n);
for(int i=1; i<=n; i++)
{
for(int j=0; j<3; j++)
Dp[j][i]=1e18;
}
Dp[0][u]=0;
Q.push({0,{0,u}});
while(!Q.empty())
{
int wh=Q.top().second.first;
int nod=Q.top().second.second;
long long cost=-Q.top().first;
Q.pop();
if(Dp[wh][nod]!=cost)
continue;
for(auto edge:A[nod])
{
if(wh==0)
{
better(0,edge.first,cost+edge.second);
if(Dist[0][nod]+edge.second+Dist[1][edge.first]==Dist[0][t] || Dist[1][nod]+edge.second+Dist[0][edge.first]==Dist[0][t])
better(1,edge.first,cost);
}
else if(wh==1)
{
if(Dist[0][nod]+edge.second+Dist[1][edge.first]==Dist[0][t] || Dist[1][nod]+edge.second+Dist[0][edge.first]==Dist[0][t])
better(1,edge.first,cost);
better(2,edge.first,cost+edge.second);
}
else
better(2,edge.first,cost+edge.second);
}
}
cout<<min(Dp[0][v],min(Dp[1][v],Dp[2][v]));
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |