제출 #76547

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
76547		Vasiljko	Commuter Pass (JOI18_commuter_pass)	C++14	100 / 100	633 ms	53704 KiB

commuter_pass

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const ll MOD = 1e9+7;
const int N =1e5+5;
const ll INF = 1e18;

int n,m,T;
bool vis[N];
ll du[N],dv[N],ds[N],dt[N],d1[N],d2[N],ans;
vector<pair<int,ll> >v[2*N];
priority_queue<pair<ll,int> >pq;
multiset<ll>su,sv;

void Dijkstra(ll *d,int s){
    for(int i=1;i<=N;i++)d[i]=INF;
    d[s]=0;
    pq.push({-d[s],s});

    while(!pq.empty()){
        auto cur=pq.top();
        pq.pop();

        for(auto e:v[cur.second]){
            if(d[cur.second]+e.second<d[e.first]){
                d[e.first]=d[cur.second]+e.second;
                pq.push({-d[e.first],e.first});
            }
        }
    }
}

void dfs(int cur){
    vis[cur]=true;

    su.insert(du[cur]);
    sv.insert(dv[cur]);
    ans=min(ans,*(su.begin())+dv[cur]);
    ans=min(ans,*(sv.begin())+du[cur]);

    for(auto e:v[cur]){
        if(ds[cur]+e.second+dt[e.first]==ds[T]&&!vis[e.first]){//if e is part of shortest path
            dfs(e.first);
        }
    }

    su.erase(su.find(du[cur]));
    sv.erase(sv.find(dv[cur]));
}

void Dijkstraa(ll *d,ll *dd)
{
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)dd[i]=d[i],pq.push({-dd[i],i});

	while(!pq.empty())
	{
		int r = pq.top().second;
		pq.pop();
		if(vis[r])continue;
		vis[r]=1;
		for(auto x:v[r])
		{
			int u = x.first;
			int c = x.second;
			if(ds[u]+c+dt[r]==ds[T] && dd[u]>dd[r])dd[u]=dd[r],pq.push({-dd[u],u});
		}
	}
}

int main()
{
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int s,t,a,b,x,y,z;

    cin>>n>>m;
    cin>>s>>t;
    cin>>a>>b;

    T=t;

    for(int i=1;i<=m;i++){
        cin>>x>>y>>z;
        v[x].push_back({y,z});
        v[y].push_back({x,z});
    }

    Dijkstra(du,a);
    Dijkstra(dv,b);
    Dijkstra(ds,s);
    Dijkstra(dt,t);
    Dijkstraa(dv,d1);
    Dijkstraa(du,d2);

    ans=INF;

    for(int i=1;i<=n;i++){
        ans=min(ans,du[i]+d1[i]);
        ans=min(ans,dv[i]+d2[i]);
    }

    cout<<ans;
    return 0;
}

• Subtask 1: 16 / 16
• Subtask 2: 15 / 15
• Subtask 3: 24 / 24
• Subtask 4: 45 / 45