HDU 3191 次短路长度和条数

#include <iostream>
#include <string>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <stack>
#include <queue>
#include <cctype>
#include <vector>
#include <iterator>
#include <set>
#include <map>
#include <sstream>
using namespace std;

#define mem(a,b) memset(a,b,sizeof(a))
#define pf printf
#define sf scanf
#define spf sprintf
#define pb push_back
#define debug printf("!\n")
#define MAXN 5010
#define MAX(a,b) a>b?a:b
#define blank pf("\n")
#define LL long long
#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x,x.begin())
#define pqueue priority_queue
#define INF 0x3f3f3f3f

struct Edge
{
    int v,w;
};
vector<Edge> g[55];
struct node
{
    int v,dist;
    int mark;//标记，0为最短路，1为次短路；
    bool operator < (const node &p) const
    {
        if(p.dist != dist)
            return p.dist<dist;

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

int n,m,s,e;
int dist[55][2];
int dp[55][2];
bool vis[55][2];
//dp[i][0]表示到达点i最短的路有多少条，dp[i][1]表示次短的条数
//dist[i][0]表示到达点i最短路的长度,dist[i][1]表示次短路的长度
/*
用v去松驰u时有四种情况 (设当前dist[v][cas])
情况1：dist[v][cas]+w(v,u)<dist[u][0]，找到一个更短的距离，则把原来最短的距离作为次短的距离，同时更新最短的.即
dist[u][1]=dist[u][0]
dist[u][0]=dist[v][cas]+w(v,u);
dp[u][1]=dp[u][0]
dp[u][0]=dp[v][cas]，
把node(dist[u][0],u,0)和node(dist[u][1],u,1)放入队列
情况2：dist[v][cas]+w(v,u)==dist[u][0]，找到一条新的相同距离的最短路，则dp[u][0]+=dp[v][cas],其他不用更新,也不入队
情况3：dist[v][cas]+w(v,u)<dist[u][1]，不可以更新最短距离，但可以更新次短的，则更新dist[u][1]和dp[u][1]
dist[u][1]=dist[v][cas]+w(v,u);
dp[u][1]=dp[v][cas];
把node(dist[u][1],u,1)放入队列
情况4：dist[v][cas]+w(v,u)==dist[u][1] 找到一条新的相同距离的次短路,则dp[u][1]+=dp[v][cas],其他不更新。
*/



void dijkstra(int start,int end)
{
    for(int i=0;i<n;i++)
    {
        dist[i][0]=dist[i][1]=INF;
    }
    memset(dp,0,sizeof(dp));
    memset(vis,false,sizeof(vis));
    priority_queue<node> Q;
    node p,q;
    dist[start][0]=0;
    dp[start][0]=1;
    p.dist=0,p.mark=0,p.v=start;
    Q.push(p);
    while(!Q.empty())
    {
        p=Q.top();
        Q.pop();
        if(vis[p.v][p.mark]) continue;
        //if(dist[p.v][p.mark]!=p.dist)continue;
        vis[p.v][p.mark]= true;
        for(int i=0;i<g[p.v].size();i++)
        {
            int v=g[p.v][i].v;
            int w=g[p.v][i].w;
            if(!vis[v][0] && p.dist+w <dist[v][0])
            {
                //可能为次短路
                if(dist[v][0]!=INF)
                {
                    q.v=v,q.dist=dist[v][1],q.mark=1;
                    dist[v][1]=dist[v][0];
                    dp[v][1]=dp[v][0];
                    Q.push(q);
                }
                dist[v][0]=p.dist+w;
                dp[v][0]=dp[p.v][p.mark];
                q.v=v,q.dist=dist[v][0],q.mark=0;
                Q.push(q);
            }
            else if(!vis[v][0] && p.dist+w==dist[v][0])
            {
                dp[v][0]+=dp[p.v][p.mark];
            }
            else if(!vis[v][1] && p.dist+w<dist[v][1])
            {
                dist[v][1]=p.dist+w;
                dp[v][1]=dp[p.v][p.mark];
                q.dist=dist[v][1],q.v=v,q.mark=1;
                Q.push(q);
            }
            else if(!vis[v][1]&&p.dist+w==dist[v][1])
            {
                dp[v][1]+=dp[p.v][p.mark];
            }
        }
    }
}



int main(){
    while(~scanf("%d%d%d%d",&n,&m,&s,&e))
    {
        for(int i=0;i<n;i++)g[i].clear();
        for(int i=1;i<=m;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            Edge p;
            p.v=v,p.w=w;
            g[u].push_back(p);
        }
        dijkstra(s,e);
        printf("%d %d\n",dist[e][1],dp[e][1]);
    }
    return 0;
}