牛客国庆集训派对Day3 I Metropolis

谁践踏了优雅 2022-05-08 11:52 146阅读 0赞

## 题目：[点击打开链接][Link 1]   题意：略。   分析：多源点最短路，只要两个不同源点的最短路相遇，我们就更新两个源点的答案。 ##

## 代码： ##

#pragma comment(linker, "/STACK:102400000,102400000")
    #include<unordered_map>
    #include<unordered_set>
    #include<algorithm>
    #include<iostream>
    #include<fstream>
    #include<complex>
    #include<cstdlib>
    #include<cstring>
    #include<cassert>
    #include<iomanip>
    #include<string>
    #include<cstdio>
    #include<bitset>
    #include<vector>
    #include<cctype>
    #include<cmath>
    #include<ctime>
    #include<stack>
    #include<queue>
    #include<deque>
    #include<list>
    #include<set>
    #include<map>
    using namespace std;
    #define pt(a) cout<<a<<endl
    #define debug test
    #define mst(ss,b) memset((ss),(b),sizeof(ss))
    #define rep(i,a,n) for (int i=a;i<=n;i++)
    #define per(i,a,n) for (int i=n-1;i>=a;i--)
    ///#define pii pair<int,int>
    #define pii pair<long long,int>
    #define fi first
    #define se second
    #define ll long long
    #define ull unsigned long long
    #define pb push_back
    #define mp make_pair
    #define inf 0x3f3f3f3f
    #define eps 1e-10
    #define PI acos(-1.0)
    const ll mod = 1e9+7;
    const int N = 2e5+10;
    
    ll qp(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
    int to[4][2]={
       {-1,0},{1,0},{0,-1},{0,1}};
    
    int n,m,p,fr[N],st[N];
    ll d[N],ans[N];
    struct eg{
        int to,len;
    };
    vector<eg> g[N];
    
    void dij() {
        for(int i=1;i<=n;i++) d[i]=2e18;
        priority_queue<pii,vector<pii>,greater<pii> > pq;
        for(int i=1;i<=p;i++) d[st[i]]=0,fr[st[i]]=st[i],pq.push({0,st[i]});
        while(!pq.empty()) {
            pii tp=pq.top(); pq.pop();
            int u=tp.se;
            if(d[u]!=tp.fi) continue;
            for(int i=0;i<g[u].size();i++) {
                eg v=g[u][i];
                if(d[u]+v.len<d[v.to]) {
                    d[v.to]=d[u]+v.len;
                    pq.push({d[v.to],v.to});
                    fr[v.to]=fr[u];
                ///更新v点最近的大都会，这时v点最近的大都会和u点相同
                }else if(fr[u]!=fr[v.to]) {
            ///v号点已经走过了而且是更优的，如果u,v两点最近的大都会不同
            ///那么假设a为u最近的大都会，b为v最近的大都会，显然d[u]是由a走到的
            ///d[v]是由b走到的，那么b有可能是a最近的大都会，a也有可能是b最近的大都会
                    ans[fr[u]]=min(ans[fr[u]],d[u]+d[v.to]+v.len);
                    ans[fr[v.to]]=min(ans[fr[v.to]],d[u]+d[v.to]+v.len);
                }
            }
        }
    }
    
    int main() {
        ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
        while(cin>>n>>m>>p) {
            for(int i=1;i<=n;i++) g[i].clear();
            for(int i=1;i<=n;i++) ans[i]=2e18;
            for(int i=1;i<=p;i++) cin>>st[i];
            for(int i=1;i<=m;i++) {
                int a,b,c;
                cin>>a>>b>>c;
                g[a].pb({b,c});
                g[b].pb({a,c});
            }
            dij();
            for(int i=1;i<p;i++) cout<<ans[st[i]]<<" ";
            cout<<ans[st[p]]<<endl;
        }
        return 0;
    }

[Link 1]: https://www.nowcoder.com/acm/contest/203/I