牛客国庆集训派对Day3 I Metropolis

题目:点击打开链接
题意:略。
分析:多源点最短路,只要两个不同源点的最短路相遇,我们就更新两个源点的答案。

代码:

#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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值