poj 3463 次短路

题意：给定一个有向图，问从起点到终点，最短路+比最短路距离长1的路的个数。

当年数据结构课程设计用A*做过，现在忘光了，2333

 #include<stdio.h>
 #include<string.h>
 #include<stdlib.h>
 #include<queue>
 #define VM 1005
 #define EM 10010
 using namespace std;
 const int inf=0x3f3f3f3f;
 int head[VM],cnt[VM][],dist[VM][],vis[VM][];//dis[i][0]为最短路，dis[i][1]为次短路
 int e,src,des,n,m;
 struct E
 {
     int to,w,next;
 } edge[EM];
 void add(int cu,int cv,int cw)
 {
     edge[e].to=cv;
     edge[e].w=cw;
     edge[e].next=head[cu];
     head[cu]=e ++;
 }
 int dij()
 {
     int i,j,u,min,flag;
     memset(dist,0x3f,sizeof(dist));
     memset(vis,,sizeof(vis));
     memset(cnt,,sizeof(cnt));
     dist[src][]=;
     cnt[src][]=;
     for(i=;i<*n;i++)
     {
         min=inf;
         for(j=;j<=n;j++)           //找新的最短路和次短路
             if(!vis[j][]&&dist[j][]<min)
             {
                 u=j;
                 flag=;
                 min=dist[j][];
             }
             else if(!vis[j][]&&dist[j][]<min)
             {
                 u=j;
                 flag=;
                 min=dist[j][];
             }
         if(min==inf)
             break;
         vis[u][flag]=;
         for(j=head[u];j!=-;j=edge[j].next)
         {
             int v=edge[j].to;
             int w=edge[j].w+min;
             if(dist[v][]>w) //如果找到的新的值比最短路小，则更新最短路和次短路的值
             {
                 dist[v][]=dist[v][];//更新次短路
                 dist[v][]=w;// 更新最短路
                 cnt[v][]=cnt[v][];
                 cnt[v][]=cnt[u][flag];//更新最短路和次短路的个数

             }
             else if(dist[v][]==w)   //如果值等于最短路
                 cnt[v][]+=cnt[u][flag];//更新最短路的个数
             else if(dist[v][]>w)    //如果找到的值小于次短路的值，更新次短路
             {
                 dist[v][]=w;    //更新次短路的值
                 cnt[v][]=cnt[u][flag];  //更新次短路的个数
             }
             else if(dist[v][]==w)   //如果找到的值等于次短路的值
                 cnt[v][]+=cnt[u][flag];//更新次短路的个数
         }

     }
     if(dist[des][]+==dist[des][])//如果次短路的值等于最短路值+1
         cnt[des][]+=cnt[des][];
     return cnt[des][];
 }
 int main()
 {
     int T,u,v,w;
     scanf("%d",&T);
     while(T--)
     {
         scanf("%d%d",&n,&m);
         memset(head,0xff,sizeof(head));
         e=;
         while(m--)
         {
             scanf("%d%d%d",&u,&v,&w);
             add(u,v,w);
         }
         scanf("%d%d",&src,&des);
         int ans=dij();
         printf("%d\n",ans);
     }
     return ;
 }