poj 3463 最短路+次短路

独立写查错不能，就是维护一个次短路的dist

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

Sample Input
2
5 8
1 2 3
1 3 2
1 4 5
2 3 1
2 5 3
3 4 2
3 5 4
4 5 3
1 5
5 6
2 3 1
3 2 1
3 1 10
4 5 2
5 2 7
5 2 7
4 1

Sample Output
3
2

2015-05-14

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 #include<queue>
 #include<map>
 using namespace std;
 #define MOD 1000000007
 const int INF=0x3f3f3f3f;
 const double eps=1e-;
 typedef long long ll;
 #define cl(a) memset(a,0,sizeof(a))
 #define ts printf("*****\n");
 const int MAXN=;
 int n,m;
 #define typec int
 int tot,src,des;
 int head[MAXN],cnt[MAXN][],dist[MAXN][],vis[MAXN][];//dis[i][0]为最短路，dis[i][1]为次短路
 struct Edge
 {
     int to,next,w;
 }edge[MAXN*];
 void addedge(int u,int v,int w)
 {
     edge[tot].to=v;
     edge[tot].w=w;
     edge[tot].next=head[u];
     head[u]=tot++;
 }
 void init()
 {
     memset(head,-,sizeof(head));
     tot=;
 }
 int dij()
 {
     int flag;
     memset(dist,0x3f,sizeof(dist));
     cl(vis);
     cl(cnt);
     dist[src][]=;
     cnt[src][]=;
     for(int j=;j<*n;j++)
     {
         int u;
         int Min=INF;
         for(int i=;i<=n;i++)
             if(!vis[i][]&&dist[i][]<Min)   //找到比最短路还短的路
             {
                 u=i;
                 flag=;
                 Min=dist[i][];
             }
             else if(!vis[i][]&&dist[i][]<Min)  //说明比最短路长，比次短路短
             {
                 u=i;
                 flag=;
                 Min=dist[i][];
             }
         if(u==-)break;
         vis[u][flag]=true;
         for(int i=head[u];i!=-;i=edge[i].next)
         {
             int v=edge[i].to;
             int w=edge[i].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 tt,u,v,w;
     #ifndef ONLINE_JUDGE
     freopen("1.in","r",stdin);
     #endif
     scanf("%d",&tt);
     while(tt--)
     {
         init();
         scanf("%d%d",&n,&m);
         while(m--)
         {
             scanf("%d%d%d",&u,&v,&w);
             addedge(u,v,w);
         }
         scanf("%d%d",&src,&des);
         int ans=dij();
         printf("%d\n",ans);
     }
     return ;
 }

2015-1-30：

 #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 ;
 }