poj 3463 次短路
题意：给定一个有向图，问从起点到终点，最短路+比最短路距离长1的路的个数。
当年数据结构课程设计用A*做过，现在忘光了，2333
1 #include<stdio.h> 2 #include<string.h> 3 #include<stdlib.h> 4 #include<queue> 5 #define VM 1005 6 #define EM 10010 7 using namespace std; 8 const int inf=0x3f3f3f3f; 9 int head[VM],cnt[VM][2],dist[VM][2],vis[VM][2];//dis[i][0]为最短路，dis[i][1]为次短路 10 int e,src,des,n,m; 11 struct E 12 { 13 int to,w,next; 14 } edge[EM]; 15 void add(int cu,int cv,int cw) 16 { 17 edge[e].to=cv; 18 edge[e].w=cw; 19 edge[e].next=head[cu]; 20 head[cu]=e ++; 21 } 22 int dij() 23 { 24 int i,j,u,min,flag; 25 memset(dist,0x3f,sizeof(dist)); 26 memset(vis,0,sizeof(vis)); 27 memset(cnt,0,sizeof(cnt)); 28 dist[src][0]=0; 29 cnt[src][0]=1; 30 for(i=1;i<2*n;i++) 31 { 32 min=inf; 33 for(j=1;j<=n;j++) //找新的最短路和次短路 34 if(!vis[j][0]&&dist[j][0]<min) 35 { 36 u=j; 37 flag=0; 38 min=dist[j][0]; 39 } 40 else if(!vis[j][1]&&dist[j][1]<min) 41 { 42 u=j; 43 flag=1; 44 min=dist[j][1]; 45 } 46 if(min==inf) 47 break; 48 vis[u][flag]=1; 49 for(j=head[u];j!=-1;j=edge[j].next) 50 { 51 int v=edge[j].to; 52 int w=edge[j].w+min; 53 if(dist[v][0]>w) //如果找到的新的值比最短路小，则更新最短路和次短路的值 54 { 55 dist[v][1]=dist[v][0];//更新次短路 56 dist[v][0]=w;// 更新最短路 57 cnt[v][1]=cnt[v][0]; 58 cnt[v][0]=cnt[u][flag];//更新最短路和次短路的个数 59 60 } 61 else if(dist[v][0]==w) //如果值等于最短路 62 cnt[v][0]+=cnt[u][flag];//更新最短路的个数 63 else if(dist[v][1]>w) //如果找到的值小于次短路的值，更新次短路 64 { 65 dist[v][1]=w; //更新次短路的值 66 cnt[v][1]=cnt[u][flag]; //更新次短路的个数 67 } 68 else if(dist[v][1]==w) //如果找到的值等于次短路的值 69 cnt[v][1]+=cnt[u][flag];//更新次短路的个数 70 } 71 72 } 73 if(dist[des][0]+1==dist[des][1])//如果次短路的值等于最短路值+1 74 cnt[des][0]+=cnt[des][1]; 75 return cnt[des][0]; 76 } 77 int main() 78 { 79 int T,u,v,w; 80 scanf("%d",&T); 81 while(T--) 82 { 83 scanf("%d%d",&n,&m); 84 memset(head,0xff,sizeof(head)); 85 e=0; 86 while(m--) 87 { 88 scanf("%d%d%d",&u,&v,&w); 89 add(u,v,w); 90 } 91 scanf("%d%d",&src,&des); 92 int ans=dij(); 93 printf("%d\n",ans); 94 } 95 return 0; 96 }
