题目:
题意:有N个点和N层..一层有X个点(0<=X<=N).两邻两层间有一条路花费C。还有M条小路在两个点之间。问从第一个点走到第N个点最短路是多少...
题解:依然是在点之间SPFA。不过在跑的时候不仅要跑与当前点相连接的点。还有把当前点所在层的相邻层的点判断是否加入队列...
CODE:
#include#include #include #include #include #include #define mkp make_pair#define fst first#define scd secondusing namespace std;int dis[100011];int vis[100011];int lay[100011];vector vec[100011];struct Edge_t{ int to,next,cap;}edge[500000];int head[100011],et;inline void adde(int u,int v,int w){ edge[et].to=v; edge[et].cap=w; edge[et].next=head[u]; head[u]=et++;}inline void spfa(int n,int C){ queue q; q.push(1); memset(vis,0,sizeof vis); memset(dis,-1,sizeof dis); dis[1]=0; int e,u,v,size,i,k; while(!q.empty()){ u=q.front();q.pop(); vis[u]=0; for(e=head[u];~e;e=edge[e].next){ v=edge[e].to; if(dis[v]<0 || dis[v]>dis[u]+edge[e].cap){ dis[v]=dis[u]+edge[e].cap; if(!vis[v]){ vis[v]=1; q.push(v); } } } if(lay[u]>1){ size=vec[k=(lay[u]-1)].size(); for(i=0;i dis[u]+C){ dis[v]=dis[u]+C; if(!vis[v]){ vis[v]=1; q.push(v); } } } } if(lay[u] dis[u]+C){ dis[v]=dis[u]+C; if(!vis[v]){ vis[v]=1; q.push(v); } } } } }}int main(){ int t,tt=0,C; int n,m,u,v,i,size,w; scanf("%d",&t); while(t--){ scanf("%d%d%d",&n,&m,&C); memset(head,-1,sizeof head);et=0; memset(vis,0,sizeof vis); for(i=1;i<=n;++i){ scanf("%d",&lay[i]);//第i个点所以层为lay[i] vec[i].clear(); } while(m--){ scanf("%d%d%d",&u,&v,&w); adde(u,v,w);adde(v,u,w); if(!vis[u]){//把点加入层.. vec[lay[u]].push_back(u); vis[u]=1; } if(!vis[v]){ vec[lay[v]].push_back(v); vis[v]=1; } } if(!vis[1])//始点和终于一定要在某一层内.. vec[lay[1]].push_back(1); if(!vis[n]) vec[lay[n]].push_back(n); for(i=1;i<=n;++i)//如果某一层没有点 if(vec[lay[i]].empty()) vec[lay[i]].push_back(i); spfa(n,C); printf("Case #%d: %d\n",++tt,dis[n]); } return 0;}