题目
约翰一共有N)个牧场.由M条布满尘埃的小径连接.小径可 以双向通行.每天早上约翰从牧场1出发到牧场N去给奶牛检查身体.
通过每条小径都需要消耗一定的时间.约翰打算升级其中K条小径,使之成为高 速公路.在高速公路上的通行几乎是瞬间完成的,所以高速公路的通行时间为0.
请帮助约翰决定对哪些小径进行升级,使他每天从1号牧场到第N号牧场所花的时间最短
代码
标准的分层图
注意要用堆优化的Dijkstra,用SPFA会T飞
#include <iostream> #include <utility> #include <cstring> #include <queue> #define mp(x,y) make_pair(x,y) using namespace std; const int N=1E5*70,M=5E7; int to[M],val[M],nxt[M],head[N],cnt; int dist[N]; bool vis[N]; int n,m,k; void add(int u,int v,int w) { to[++cnt]=v; val[cnt]=w; nxt[cnt]=head[u]; head[u]=cnt; } void dijkstra() { //这题卡SPFA,只能用堆优化的Dijkstra memset(dist,0x3f,sizeof(dist)); dist[1]=0; priority_queue<pair<int,int> > q; q.push(mp(0,1)); while(!q.empty()) { int x=q.top().second; q.pop(); vis[x]=true; for(int i=head[x];i;i=nxt[i]) { int y=to[i],w=val[i]; if(dist[y]>dist[x]+w) { dist[y]=dist[x]+w; if(!vis[y]) q.push(mp(-dist[y],y)); } } } } int main() { cin>>n>>m>>k; while(m--) { int u,v,w; cin>>u>>v>>w; add(u,v,w); add(v,u,w); for(int j=1;j<=k;j++) { add(v+j*n,u+j*n,w); add(u+j*n,v+j*n,w); add(v+j*n-n,u+j*n,0); add(u+j*n-n,v+j*n,0); } } dijkstra(); int ans=0x7fffffff; //防极品数据 for(int i=1;i<=k;i++) ans=min(ans,dist[i*n+n]); cout<<ans; return 0; }
来源:https://www.cnblogs.com/wyc06/p/12641813.html