此题可用单源最短路径来做,不过路径计算为2*d+w。配合优先队列的dij可以不超时地实现。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
struct point
{
int id;
long long price;
};
struct road //边
{
int y,nt;
long long d;
};
bool operator <(point a,point b) //自定义排序方式,从小到大
{
return a.price>b.price;
}
const int maxn=2e5+7;
int head[maxn];
bool vis[maxn];
road f[maxn*2]; //边集数组
long long dis[maxn]; //需注意longlong,下标处的最低票价
int tot=0;
void add(int tx,int ty,long long td) //编辑数组加边
{
tot++; f[tot].nt=head[tx]; head[tx]=tot; f[tot].y=ty; f[tot].d=td;
}
int main()
{
priority_queue<point> q; //优先队列优化dij
int n,m,i,tx,ty;
long long td;
point t,tmp;
memset(head,-1,sizeof(head));
memset(vis,false,sizeof(vis));
scanf("%d%d",&n,&m);
for(i=0;i<m;i++)
{
scanf("%d%d%lld",&tx,&ty,&td);
add(tx,ty,td);
add(ty,tx,td);
}
for(i=1;i<=n;i++)
{
scanf("%lld",&dis[i]); //dis初值为在本城看
t.id=i; t.price=dis[i];
q.push(t);
}
while(!q.empty())
{
tmp=q.top();
if(vis[tmp.id])
{
q.pop(); continue; //dij,优先队列优化后每个点最多扩一次
}
vis[tmp.id]=true;
i=head[tmp.id];
while(i!=-1)
{
if(dis[f[i].y]>dis[tmp.id]+f[i].d*2)
{
dis[f[i].y]=dis[tmp.id]+f[i].d*2; //核心,j点最优后,扩展出的点与原来比较,便可之间赋值。可看成先到j,即变成j点出发。
t.id=f[i].y; t.price=dis[f[i].y];
q.push(t);
}
i=f[i].nt;
}
q.pop();
}
for(i=1;i<n;i++) printf("%lld ",dis[i]);
printf("%lld\n",dis[n]);
return 0;
}
CodeForces - 938D Buy a Ticket
最新推荐文章于 2020-03-31 19:05:41 发布