POJ3463 最短路和次短路路径的条数（好题）

题目链接：http://poj.org/problem?id=3463

题目大意及分析：

　　（我这题不会，纯属抄袭）

　　http://www.cppblog.com/Yuan/archive/2010/04/19/112983.html

 

代码风格不错，学习了下dijkstra+优先队列来解决这种题目，也学到了那个Node(int v_, int f_){v=v_,f=f_;}的写法；

 

代码：

/*3463    Accepted    572K    47MS    C++    2481B    2012-06-23 19:44:41*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

#define mpair make_pair
#define pii pair<int,int>
#define MM(a,b) memset(a,b,sizeof(a));
typedef long long lld;
typedef unsigned long long u64;
template<class T> bool up_max(T& a,const T& b){return b>a? a=b,1 : 0;}
template<class T> bool up_min(T& a,const T& b){return b<a? a=b,1 : 0;}
#define maxn 1010
const int inf= 2100000000;

int n,m;
vector< pii > map[maxn];

bool vis[maxn][2];
int dis[maxn][2];
int cnt[maxn][2];

struct Node{
    int v,f;
    Node(int v_,int f_){v=v_,f=f_;}
    bool operator<(Node a)const{
        return dis[v][f] > dis[a.v][a.f];
    }
};

priority_queue<Node>Q;
void Dijkstra(int st, int ed){
    for(int i=1;i<=n;++i){
        vis[i][0]= vis[i][1]= 0;
        cnt[i][0]= cnt[i][1]= 0;
        dis[i][0]= dis[i][1]= inf;
    }
    while( !Q.empty() ) Q.pop();
    dis[st][0]= 0, cnt[st][0]= 1;
    Q.push( Node( st, 0 ) );
    while( !Q.empty() ){
        Node a= Q.top(); Q.pop();
        int u= a.v, f= a.f;
        if( vis[u][f] ) continue;
        vis[u][f]= 1;
        for(int i=0;i<map[u].size();++i){
            int v= map[u][i].first, w= dis[u][f]+map[u][i].second;
            if( w<dis[v][0] ){
                if( dis[v][0]!=inf ){
                    dis[v][1]= dis[v][0];
                    cnt[v][1]= cnt[v][0];
                    Q.push( Node( v, 1 ) );
                }
                dis[v][0]= w;
                cnt[v][0]= cnt[u][f];
                Q.push( Node( v, 0 ) );
            }
            else if( w==dis[v][0] ) cnt[v][0]+= cnt[u][f];
            else if( up_min( dis[v][1], w ) ){
                cnt[v][1]= cnt[u][f];
                Q.push( Node( v, 1 ) );
            }
            else if( w==dis[v][1] ) cnt[v][1]+= cnt[u][f];
        }
    }
}

int main()
{
    //freopen("poj3463.in","r",stdin);
    int Cas, i, j, x, y, t;
    cin>>Cas;
    while( Cas-- ){
        cin>>n>>m;
        for(i=1;i<=n;++i) map[i].clear();
        for(i=1;i<=m;++i){
            scanf("%d%d%d", &x, &y, &t);
            map[x].push_back( pii( y, t ) ); //unidirectional;
        }
        int st, ed;
        scanf("%d%d",  &st, &ed);
        Dijkstra( st, ed );
        int ans= cnt[ed][0];
        if( dis[ed][0]==dis[ed][1]-1 ) ans+= cnt[ed][1];
        cout<< ans <<endl;
    }
}