이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define i128 int128
#define ii pair<int,int>
#define ld long double
#define vi vector<int>
#define vvi vector<vi>
#define vii vector<pair<int,int>>
#define FOR(i,a,b) for (int i=a;i<=b;i++)
#define FOR1(i,a,b,c) for (int i=a;i<=b;i+=c)
#define REP(i,a,b) for (int i=a;i>=b;i--)
#define REP1(i,a,b,c) for(int i=b;i>=a;i-=c)
#define endh '\n';
#define len(a) a.length()
#define pb(c) push_back(c)
#define elif else if
#define ppb() pop_back()
#define rong std::npos
#define all(c) (c.begin(),c.end())
#define Z int
#define R double
#define lcm(a,b) ((int) (a/__gcd(a,b))*b)
#define mk(a,b) make_pair(a,b)
Z dx4[]={-1,1,0,0};
Z dy4[]={0,0,1,-1};
string co="YES",khong="NO";
const int mod=1e9+7;
const Z maxn=200005;
Z n,m,s,t,toi,kurumi;
Z d1[maxn],d2[maxn],d3[maxn];
vii g[maxn];
void dijkstra1()
{
memset(d1,0x3f3f3f3f,sizeof(d1));
d1[s]=0;
priority_queue<ii,vii,greater<ii>>q;
q.push(mk(0,s));
while(!q.empty()){
ii top=q.top();q.pop();
Z u=top.second;
Z w=top.first;
if (w>d1[u]) continue;
for (ii i:g[u]){
Z u1=i.first;
Z w1=i.second;
if (d1[u]+w1<d1[u1]){
d1[u1]=d1[u]+w1;
q.push(mk(d1[u1],u1));
}
}
}
}
void dijkstra2()
{
memset(d2,0x3f3f3f3f,sizeof(d2));
d2[t]=0;
priority_queue<ii,vii,greater<ii>>q;
q.push(mk(0,t));
while(!q.empty()){
ii top=q.top();q.pop();
Z u=top.second;
Z w=top.first;
if (w>d2[u]) continue;
for (ii i:g[u]){
Z u1=i.first;
Z w1=i.second;
if (d2[u]+w1<d2[u1]){
d2[u1]=d2[u]+w1;
q.push(mk(d2[u1],u1));
}
}
}
}
void dijkstra3()
{
memset(d3,0x3f3f3f3f,sizeof(d3));
d3[toi]=0;
priority_queue<ii,vii,greater<ii>>q;
q.push(mk(0,toi));
while(!q.empty()){
ii top=q.top();q.pop();
Z u=top.second;
Z w=top.first;
if (u==kurumi){
cout<<w;
return;
}
if (w>d2[u]) continue;
for (ii i:g[u]){
Z u1=i.first;
Z w1=i.second;
if ( d2[u]+d1[u1]+w1==d1[t]) w1=0;
if (d3[u]+w1<d3[u1]){
d3[u1]=d3[u]+w1;
q.push(mk(d3[u1],u1));
}
}
}
}
signed main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin>>n>>m>>s>>t;
cin>>toi>>kurumi;
FOR(i,1,m){
Z u,v,w;
cin>>u>>v>>w;;
g[u].pb(mk(v,w));
g[v].pb(mk(u,w));
}
dijkstra1();
dijkstra2();
dijkstra3();
}
# | 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... |