题意:给出一个图,求从1到n的最短路,另外,每个结点属于一个层,层数相差为1的两个结点可以相互传送,距离固定为c。
思路:基本算法就是最短路,不管是dijkstra或者spfa都行吧, 我用的是djkstra……主要问题就是构图,当时做热身赛的时候死活没想出来,我果然还是太嫩了么。。。后来洗衣服的时候突然想到一个自以为不错的构图思路,赶紧跑回去写……当时想到的思路是这样的:在相隔为1的两个层之间设一个虚点,然后两个层之间的点分别连向这个虚点,一条边权值为c,另一条为0,结果交完还是wa,后来发现从某一层的点可以通过这个虚点再到达这层的另一个点,知道了这个问题其实就比较好改了,还是这个思路,每一层之间设两个虚点,一个从低一层到高一层,另一个从高一层到低一层,注意这里连的是有向边,这样的话点的个数是n*3,边的个数是n*5,跑一边模板就基本就没问题了~
代码:
#include<iostream> #include<cstdio> #include<cstring> #include<string> #include<algorithm> #include<map> #include<queue> #include<set> #include<stack> #include<cmath> #include<vector> #define inf 0x3f3f3f3f #define Inf 0x3FFFFFFFFFFFFFFFLL #define eps 1e-9 #define pi acos(-1.0) using namespace std; typedef long long ll; const int maxn=100000+10; struct Edge { int v,w; int next; }; struct HeapNode { int d,u; bool operator < (const HeapNode &a) const { return d>a.d; } }; Edge edges[maxn*6]; int head[maxn<<2],nEdge,n,m,c; int d[maxn<<2]; bool vis[maxn<<2]; vector<int>layer[maxn]; void AddEdge(int u,int v,int w) { nEdge++; edges[nEdge].v=v; edges[nEdge].w=w; edges[nEdge].next=head[u]; head[u]=nEdge; } void djkstra(int N) { priority_queue<HeapNode>q; memset(vis,0,sizeof(vis)); for(int i=1;i<=N;++i) d[i]=inf; d[1]=0; HeapNode hp; hp.u=1;hp.d=0; q.push(hp); while(!q.empty()) { hp=q.top();q.pop(); int u=hp.u; if(vis[u]) continue; vis[u]=true; for(int k=head[u];k!=-1;k=edges[k].next) { int v=edges[k].v; if(d[v]>d[u]+edges[k].w) { d[v]=d[u]+edges[k].w; hp.d=d[v];hp.u=v; q.push(hp); } } } } int main() { //freopen("in.txt","r",stdin); //freopen("out.txt","w",stdout); int t,tcase=0; scanf("%d",&t); while(t--) { tcase++; int u,v,w,tmp; scanf("%d%d%d",&n,&m,&c); memset(head,0xff,sizeof(head)); nEdge=-1; for(int i=0;i<=n;++i) layer[i].clear(); for(int i=1;i<=n;++i) { scanf("%d",&tmp); layer[tmp].push_back(i); } for(int i=0;i<m;++i) { scanf("%d%d%d",&u,&v,&w); AddEdge(u,v,w); AddEdge(v,u,w); } int sz; for(int i=1;i<n;++i) { sz=layer[i].size(); for(int j=0;j<sz;++j) { AddEdge(layer[i][j],i+n,c); AddEdge(i+n+n,layer[i][j],0); } sz=layer[i+1].size(); for(int j=0;j<sz;++j) { AddEdge(i+n,layer[i+1][j],0); AddEdge(layer[i+1][j],i+n+n,c); } } djkstra(n*3); printf("Case #%d: ",tcase); if(d[n]==inf||n==0) printf("-1\n"); else printf("%d\n",d[n]); } return 0; }