HDU-1688 Sightseeing 最短路与“次短路”条数

http://acm.hdu.edu.cn/showproblem.php?pid=1688

思路：http://blog.csdn.net/u011742541/article/details/13025079

#include "stdio.h"
#include "string.h"
#include "vector"
#include "queue"
using namespace std;
const int maxn = 1005;
const int inf = 1<<30;
int n,m,s,e;
int dis[maxn][2],cnt[maxn][2],vis[maxn][2]; 
//dis[i][0]表示到达点i最短路的长度,dis[i][1]表示次短路的长度 
//cnt[i][0]表示到达点i最短路的条数,cnt[i][1]表示次短路的条数 
struct node
{
    int to,w;
    node( int a,int b ) { to = a; w = b; }
};
vector<node>map[maxn];

struct Node
{
    int v,dis;
    int mark;    //标记最短还是次短
    bool operator < ( const Node &p ) const 
    {
         if( p.dis != dis )
            return p.dis < dis;

         return p.v < v;//这儿如果不按顶点的大小排序，就wa了。
    }
};

void Dijstra( int s,int e )
{
    for( int i = 1; i <= n; i ++ )
    {
        dis[i][0] = dis[i][1] = inf;
        cnt[i][0] = cnt[i][1] = 0;
    }
    dis[s][0] = 0;  cnt[s][0] = 1;
    memset( vis,0,sizeof(vis) );
    priority_queue<Node>que;
    Node u,v;
    u.dis = 0; u.mark = 0; u.v = s;
    que.push( u );
    while( !que.empty())
    {
        u = que.top();  que.pop();
        if( vis[u.v][u.mark] )
            continue;
        vis[u.v][u.mark] = true;
        //松驰时有四种情况
        for( int i = 0;i < map[u.v].size(); i ++ )
        {
            node x = map[u.v][i];
            if( !vis[x.to][0] && dis[x.to][0] > u.dis + x.w )  //找到一个更短的距离，则把原来最短的距离作为次短的距离，同时更新最短的.
            {
                if( dis[x.to][0] != inf )                       //可能为次短路
                {
                    v.v = x.to; v.dis = dis[x.to][0]; v.mark = 1;
                    dis[x.to][1] = dis[x.to][0]; 
                    cnt[x.to][1] = cnt[x.to][0]; 
                    que.push( v );
                }
                dis[x.to][0] = u.dis + x.w;
                cnt[x.to][0] = cnt[u.v][u.mark];
                v.v = x.to; v.dis = dis[x.to][0]; v.mark = 0;
                que.push( v );
            }
            else if( !vis[x.to][0] && dis[x.to][0] == u.dis + x.w )
                cnt[x.to][0] += cnt[u.v][u.mark];

            else if( !vis[x.to][1] && dis[x.to][1] > u.dis + x.w )//不可以更新最短距离，但可以更新次短的
            {
                dis[x.to][1] = u.dis + x.w;
                cnt[x.to][1] = cnt[u.v][u.mark];
                v.v = x.to; v.dis = dis[x.to][1]; v.mark = 1;
                que.push( v );
            }
            else if( !vis[x.to][1] && dis[x.to][1] == u.dis + x.w )//找到一条新的相同距离的次短路
                cnt[x.to][1] += cnt[u.v][u.mark];
        }
    }
}

int main()
{
    int t,a,b,c;
    //freopen( "data.txt","r",stdin );
    scanf("%d",&t);
    while( t-- )
    {
        scanf("%d%d",&n,&m);
        for( int i = 1; i <= n; i ++ )
            map[i].clear();
        for( int i = 1; i <= m; i ++ )
        {
            scanf("%d%d%d",&a,&b,&c);
            map[a].push_back( node(b,c) );
        }
        scanf("%d%d",&s,&e);
        Dijstra( s,e );
        if( dis[e][1] == dis[e][0] + 1 )
            printf("%d\n",cnt[e][0]+cnt[e][1]);
        else
            printf("%d\n",cnt[e][0]);
    }
    return 0;
}