Educational Codeforces Round 38-D- Buy a Ticket(SPFA)
程序员文章站
2022-06-27 11:51:58
...
题意:n个城市,每个城市由一个人,每个人都想看一场演唱会,并且n个城市都将举办演唱会,但是每个城市演唱会的门票价格不同,一个人可以在自己城市看也可以去能到达的城市看,若第i个城市的人去第j个城市(假设能到达)看的话,会花费dis(i,j)*2+aj的花费,其中aj表示j城市演唱会的门票价格。现在有m条边描述了若干个城市的联通情况,问你每个人看演唱会花费的最小代价分别为多少?
题解:我们将n个城市分别和原点建边,建立原点为n+1,建边的权值为ai,将原点放进队列,跑spfa,最后更新出的dis数组就是答案。
#include<queue>
#include<vector>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll long long
#define maxn 200005
#define inf ((1ll<<60-1))
int n,m,x,y,vis[maxn];
ll z,ans[maxn],dis[maxn];
priority_queue<pair<ll,int> > t;
vector<pair<int,ll> >q[maxn];
void work(int len,int st)
{
for(int i=1;i<len;i++)
dis[i]=inf;
t.push(make_pair(dis[st]=0,st));
while(t.empty()==0)
{
int now=t.top().second;
t.pop();vis[now]=0;
for(int i=0;i<q[now].size();i++)
{
int v=q[now][i].first;
ll c=q[now][i].second;
if(dis[v]>dis[now]+c)
{
dis[v]=dis[now]+c;
if(!vis[v])
vis[v]=1,t.push(make_pair(-dis[v],v));
}
}
}
}
int main(void)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d%lld",&x,&y,&z);
q[x].push_back(make_pair(y,2ll*z));
q[y].push_back(make_pair(x,2ll*z));
}
for(int i=1;i<=n;i++)
{
scanf("%lld",&z);
q[n+1].push_back(make_pair(i,z));
}
work(n+1,n+1);
for(int i=1;i<=n;i++)
printf("%lld ",dis[i]);
printf("\n");
return 0;
}