HDU 1688 Sightseeing 最短路 及次短路 路径长度和路径数

本文来自：　http://blog.csdn.net/napoleon_acm/article/details/24583767

题目链接： http://acm.hdu.edu.cn/showproblem.php?pid=1688

题目： 给定一有向图 点和边个数 2 ≤ N ≤ 1,000 and 1 ≤ M ≤ 10, 000， 给定 起点和终点，求 最短路 和比最短路大1个单位 的路径的 路径数之和。

首先要会Dijkstra 求最短路， 之后修改一下就可以了。

dis[i][2],dp[i][2] 分别表示 起点到i点的 最短 及次短路径长度，以及方案数。

求最短路是dijkstra 算法 是循环n次 一次更新每个点的最短距离，  现在求最短以及次短就要 循环2*n次。 

对于每次更新： 

1.新路径比最短路径长度要小，那么最短路和次短路的长度和次数都要更新。
2.新路径等于最短路的长度，那么只需要更新最短路的条数。
3.新路径比最短路要长但是比次短路要短，那么需要更新次短路的长度和条数。
4.新路径等于次短路径的长度，那么只需要更新次短路径的条数。

之后判断终点的 次短路是否比最短路大1就可以了。 

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define INF 1000000000
#define N 1005
int t,tot,hh[N],n,m,dp[N][3],dis[N][3],S,T,vis[N][3];
struct node
{
    int v,next,w;
}edge[100000];
void init()
{
    memset(hh,-1,sizeof(hh));
    tot=0;
}
void add(int u,int v,int w)
{
    edge[tot].v=v; edge[tot].w=w;
    edge[tot].next=hh[u]; hh[u]=tot++;
}
void dijk()
{
    memset(dp,0,sizeof(dp));
    for(int i=0;i<=n;i++) dis[i][0]=dis[i][1]=INF;
    memset(vis,0,sizeof(vis));
    dp[S][1]=1;
    dis[S][1]=0; //不要先进行对起点的更新，因为可能有重边
    for(int q=1;q<=2*n;q++)
    { 
        int ma,ad,type;
        ma=INF; ad=-1; type=-1;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i][1] && ma>dis[i][1])
            {
                ad=i; ma=dis[i][1]; type=1;
            }
            else if(!vis[i][0] && ma>dis[i][0])
            {
                ad=i; ma=dis[i][0]; type=0;
            }
        }
        if(ad==-1) break;
        vis[ad][type]=1;
        for(int i=hh[ad];i!=-1;i=edge[i].next)
        {
            int cost=edge[i].w;
            int vv=edge[i].v;
            if(dis[vv][1]>ma+cost)
            {
                dis[vv][0]=dis[vv][1];
                dp[vv][0]=dp[vv][1];
                dis[vv][1]=ma+cost;
                dp[vv][1]=dp[ad][type];
            }
            else if(dis[vv][1]==ma+cost)
                dp[vv][1]+=dp[ad][type];
            else if(dis[vv][0]>ma+cost)
            {
                dis[vv][0]=ma+cost;
                dp[vv][0]=dp[ad][type];
            }
            else if(dis[vv][0]==ma+cost)
                dp[vv][0]+=dp[ad][type];
        }
    }
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen ("in.txt" , "r" , stdin);
#endif
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d%d",&n,&m);
        int u,v,w;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        scanf("%d%d",&S,&T);
        dijk();
        int sum=dp[T][1];
        if(dis[T][1]+1==dis[T][0]) sum+=dp[T][0];
        printf("%d\n",sum);
    }
    return 0;
}