POJ 3463 Sightseeing 次短路条数

题目大意：给一个图，求起点到终点的的最短路条数加上最短路+1条数的和。

这个问题其实可以分解为，求最短路、最短路条数、次短路、次短路条数。

第一个问题很简单，就不谈了。

第二个问题这里要说一下，求条数，我们有个简单想法是开另外一个数组（ 比方cnt[ ] ）记录条数。还是求最短路。在计算中。用u更新v，如果能把到v的最短路改小，这时cnt[v]=cnt[u]，若是更新的路和最短路相等，这时，cnt[v]+=cnt[u]。思想很简单，但是对于spfa来说，很麻烦，很容易出bug（实际上，我从来没用spfa写过 = =！）。但是dij则不会出现这个问题，因为他找出的最短路，一定是真正意义上的最短，其他点不能更新这个点了。

以这个图为例，1为起点，6为终点。假设用spfa，最开始，队内只有1点，路径数为1，假设先更新4，再更新2。更新后，队内会有两个点，其中4在队头，2在队尾，路径数都是1。继续spfa的过程，你会发现到6的路径数会是3 o(╯□╰)o。但是，dij不会有这个问题。

最后还有问题3和4，其实问题3就是把最短和次短认为是两个有限制的点罢了，而问题4和问题2同理。

为了省一个标记，我把点x的最短认为x*2，次短认为x*2+1

//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<cctype>
#include<string>
#include<algorithm>
#include<iostream>
#include<ctime>
#include<map>
#include<set>
using namespace std;
#define MP(x,y) make_pair((x),(y))
#define PB(x) push_back(x)
typedef long long LL;
//typedef unsigned __int64 ULL;
/* ****************** */
const int INF=1000111222;
const double INFF=1e200;
const double eps=1e-8;
const LL mod=1000000007;
const int NN=1100;
const int MM=10010;
/* ****************** */

struct G
{
    int v,len,next;
}E[MM];
int p[NN],T;
bool vis[NN*2];
int dis[NN*2],cnt[NN*2];

struct node
{
    int x,dis;
    node(int a=0,int b=0)
    {
        x=a;
        dis=b;
    }
    friend bool operator< (node t1,node t2)
    {
        return t1.dis>t2.dis;
    }
};

void add(int u,int v,int len)
{
    E[T].v=v;
    E[T].len=len;
    E[T].next=p[u];
    p[u]=T++;
}

void dij(int st,int n)
{
    int i,v;
    node ix=node(st<<1,0);
    priority_queue< node >q;
    for(i=1;i<=n+n+1;i++)
    {
        vis[i]=false;
        dis[i]=INF;
    }
    dis[st<<1]=0;
    cnt[st<<1]=1;
    q.push(ix);

    while(!q.empty())
    {
        ix=q.top();
        q.pop();
        if( vis[ ix.x ] )
            continue;

    //    printf("%d %d cnt==%d\n",ix.x,dis[ix.x],cnt[ix.x]);

        vis[ ix.x ]=true;
        for(i=p[ ix.x>>1 ];i+1;i=E[i].next)
        {
            v=E[i].v;
            if(dis[v<<1]>dis[ix.x]+E[i].len)
            {
                cnt[v<<1|1]=cnt[v<<1];
                dis[v<<1|1]=dis[v<<1];
                dis[v<<1]=dis[ix.x]+E[i].len;
                cnt[v<<1]=cnt[ix.x];

                q.push( node(v<<1,dis[v<<1]) );
                q.push( node(v<<1|1,dis[v<<1|1]) );
            }
            else if(dis[v<<1]==dis[ix.x]+E[i].len)
            {
                cnt[v<<1]+=cnt[ix.x];
            }
            else if(dis[v<<1|1]>dis[ix.x]+E[i].len)
            {
                dis[v<<1|1]=dis[ix.x]+E[i].len;
                cnt[v<<1|1]=cnt[ix.x];

                q.push( node(v<<1|1,dis[v<<1|1]) );
            }
            else if(dis[v<<1|1]==dis[ix.x]+E[i].len)
            {
                cnt[v<<1|1]+=cnt[ix.x];
            }
        }
    }
}

int main()
{
    int cas;
    int n,m,st,en;
    int u,v,len;
    scanf("%d",&cas);
    while(cas--)
    {
        memset(p,-1,sizeof(p));
        T=0;

        scanf("%d%d",&n,&m);
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&len);
            add(u,v,len);
        }
        scanf("%d%d",&st,&en);
        dij(st,n);
        if(dis[en<<1]+1==dis[en<<1|1])
        {
            printf("%d\n",cnt[en<<1]+cnt[en<<1|1]);
        }
        else
        {
            printf("%d\n",cnt[en<<1]);
        }
    }
    return 0;
}