Submission #1274021

#	Time	Username	Problem	Language	Result	Execution time	Memory
1274021		quanduongxuan12	Commuter Pass (JOI18_commuter_pass)	C++20	100 / 100	242 ms	21472 KiB

commuter_pass

#include <bits/stdc++.h>
using namespace std;
#define name "commuter"
#define MAXN 200005
#define pb push_back
#define pf push_front
#define ll long long
#define ii pair<int, int>
#define fs first
#define sc second
#define ill pair<int, ll>
#define lli pair<ll, int>
#define llll pair<ll, ll>
#define all(v) v.begin(),v.end()
#define uni(v) v.erase(unique(all(v)),v.end())
#define bit(n,i) (((n)>>(i))&1)
#define FOR(i,a,b) for (int i=(a),_b=(b); i<=_b; i++)
#define FORD(i,a,b) for (int i=(a),_b=(b); i>=_b; i--)
#define MASK(i) (1LL<<(i))
const ll INF=1e18;
const int MOD=1e9+7;
void add(int &u, int v){
    u+=v;
    if (u>=MOD) u-=MOD;
}
void sub(int &u, int v){
    u-=v;
    if (u<0) u+=MOD;
}
void minimize(ll &u, ll v){
    u=min(u,v);
}
void maximize(int &u, int v){
    u=max(u,v);
}
long long Rand(long long l, long long r){
    ll tmp=0;
    FOR(i,1,4) tmp=((tmp<<15)^(((1<<15)-1)&rand()));
    return l+tmp%(r-l+1);
}
int n,m;
vector<ii> adj[MAXN];
vector<int> child[MAXN],par[MAXN];
int S,T,U,V;
ll d[3][MAXN];
bool visited[MAXN];
priority_queue<lli,vector<lli>,greater<lli>> pq;
void dijkstra(int type, int s){
    FOR(i,1,n) d[type][i]=INF;
    d[type][s]=0;
    pq.push({0,s});
    while (pq.size()){
        lli tmp=pq.top();
        pq.pop();
        int u=tmp.sc;
        if (visited[u]) continue;
        visited[u]=1;
        for (ii pairs:adj[u]){
            int v=pairs.fs;
            int wei=pairs.sc;
            if (d[type][v]>d[type][u]+1LL*wei){
                d[type][v]=d[type][u]+1LL*wei;
                pq.push({d[type][v],v});
            }
        }
    }
}
ll ans;
ll f[MAXN],g[MAXN];
void dfs(int u){
    f[u]=d[1][u];
    g[u]=d[2][u];
    visited[u]=1;
    for (ii pairs:adj[u]){
        int v=pairs.fs;
        int w=pairs.sc;
        if (d[0][v]+1LL*w==d[0][u]){
            if (!visited[v]) dfs(v);
            minimize(f[u],f[v]);
            minimize(g[u],g[v]);
        }
    }
    minimize(ans,min(f[u]+d[2][u],g[u]+d[1][u]));
}
struct edge{
    int u,v,w;
}e[MAXN];
int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin>>n>>m;
    cin>>S>>T;
    cin>>U>>V;
    FOR(i,1,m) cin>>e[i].u>>e[i].v>>e[i].w;
    FOR(i,1,m){
        int u=e[i].u,v=e[i].v,w=e[i].w;
        adj[u].pb({v,w});
        adj[v].pb({u,w});
    }
    dijkstra(0,S);
    FOR(i,1,n) visited[i]=0;
    dijkstra(1,U);
    FOR(i,1,n) visited[i]=0;
    dijkstra(2,V);
    FOR(i,1,n) visited[i]=0;
    ans=d[1][V];
    dfs(T);
    cout<<ans;
    return 0;
}

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