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 }