HDU - 1688 Sightseeing 最短路和次短路计数

HDU 1688

不是很难的一个题，因为只是求次短路，用k短路求就有点大材小用了，所以可以用dijkstra求次短路。
只需要在原来松弛的时候进行改动：
(1) 当前权值比最短路小，那么把当前最短路的数据复制到次短路，让后更新最短路
(2) 当前权值等于最短路，那么直接将次数加起来
(3) 当前权值比次短路小，那么直接更新次短路
(4) 当前权值等于次短路，那么直接将次数加起来

wa了20多发，照网上代码改了好几个，一模一样都能wa掉，结果发现是我自己写的变量有歧义了。。测试数据用的t，终点也用的t，直接当场去世。以后还是用 _ 这样的当变量吧~~~

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define pb push_back
#define mk make_pair
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int N=1010,M=100010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,m,s,t;
int e[M],ne[M],w[M],h[N*2],idx;
int dis[2][N],cnt[2][N];
bool st[2][N];
struct Node
{
    int u,w,f;
	friend bool operator < (Node a , Node b)
	{
		return a.w>b.w;
	}
};

void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void init()
{
    idx=0;
    memset(cnt,0,sizeof (cnt));
    memset(st,false,sizeof (st));
    memset(h,-1,sizeof (h));
    for(int i=1;i<=n;i++) dis[1][i]=dis[0][i]=2e9+7;
}

void dijkstra()
{
    priority_queue<Node>q; q.push({s,0,0});
    dis[0][s]=0; cnt[0][s]=1;
    while(q.size())
    {
        Node t=q.top(); q.pop();
        int u=t.u,dd=t.w,p=t.f;
        if(st[p][u]) continue;
        st[p][u]=true;
        for(int i=h[u];~i;i=ne[i])
        {
            int j=e[i]; int d=dis[p][u]+w[i];
            if(dis[0][j]>dis[p][u]+w[i])
            {
                dis[1][j]=dis[0][j]; cnt[1][j]=cnt[0][j];
                dis[0][j]=dis[p][u]+w[i]; cnt[0][j]=cnt[p][u];
                q.push({j,dis[0][j],0}); q.push({j,dis[1][j],1});
            }
            else if(dis[0][j]==dis[p][u]+w[i])
                cnt[0][j]+=cnt[p][u];
            else if(dis[1][j]>dis[p][u]+w[i])
            {
                dis[1][j]=dis[p][u]+w[i];
                cnt[1][j]=cnt[p][u];
                q.push({j,dis[1][j],1});
            }
            else if(dis[1][j]==dis[p][u]+w[i])
                cnt[1][j]+=cnt[p][u];
        }
    }
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    int _; scanf("%d",&_);
    while(_--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=1;i<=m;i++)
        {
            int a,b,c; scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
        }
        scanf("%d%d",&s,&t);
        dijkstra();
        if(dis[1][t]==dis[0][t]+1) printf("%d\n",cnt[1][t]+cnt[0][t]);
        else printf("%d\n",cnt[0][t]);
    }

	return 0;
}
/*
4 4
1 2 1
1 3 2
2 4 1
3 4 2
1 4
*/