hdu-1688次短路问题

最近才听说次短路问题，以为挺简单其实是眼高手低，

题目意思就是裸的次短路，问最短路的个数和与最短路差1的个数和，考虑dijkstra的特性，可以利用每次进队列的方式方式实现。

朴素的dij就是进dis数列中的最小的，还要标记vis，那么次短路可以看做分类讨论

首先开出最短路和次短路的dis，cnt，和vis，那么入队方式什么的都和朴素的一样，

但是注意分类讨论，

如果拿出来的是个最短路，就分情况看看是不是影响最短路，如果影响最短路，那么次短路也相应转移，所以把两个都入，

如果不影响最短路，再判断是不是影响次短路，同理。

如果拿出来的是个次短路，就分情况看看是不是影响次短路，同理。

如果说dij是，似乎有些dp思想。。。

太弱了，最近还习惯性手残。。。。继续努力。。。加油

#include <iostream>#include <vector>#include <stdio.h>#include <queue>#include <cstring>#include <math.h>using namespace std;const int maxn=100005;int n,m,x,y;int dis[maxn][2];int cnt[maxn][2];int vis[maxn][2];struct edge{    int x,val;    edge(int a,int c):x(a),val(c){};};struct node{    int x,y,val;    node(int a,int b,int c):x(a),y(b),val(c){};};bool operator <(const node e1,const node e2){    return e1.val>e2.val;}vector<edge> v[maxn];void dij(int x,int y){    for(int i=0;i<=n;i++)        dis[i][0]=dis[i][1]=1<<30;    memset(vis,0,sizeof(vis));    memset(cnt,0,sizeof(cnt));    priority_queue<node> pq;    dis[x][0]=0;    cnt[x][0]=1;    pq.push(node(x,0,0));    while(!pq.empty()){        node cs=pq.top();        pq.pop();        if(vis[cs.x][cs.y])continue;        vis[cs.x][cs.y]=1;        for(int i=0;i<v[cs.x].size();i++){            if(cs.y==0){                if(dis[cs.x][0]+v[cs.x][i].val<=dis[v[cs.x][i].x][0]){                    if(dis[cs.x][0]+v[cs.x][i].val==dis[v[cs.x][i].x][0]){                        cnt[v[cs.x][i].x][0]+=cnt[cs.x][0];                    }                    else{                        cnt[v[cs.x][i].x][1]=cnt[v[cs.x][i].x][0];                        cnt[v[cs.x][i].x][0]=cnt[cs.x][0];                        dis[v[cs.x][i].x][1]=dis[v[cs.x][i].x][0];                        pq.push(node(v[cs.x][i].x,1,dis[v[cs.x][i].x][1]));                    }                    dis[v[cs.x][i].x][0]=dis[cs.x][0]+v[cs.x][i].val;                    pq.push(node(v[cs.x][i].x,0,dis[v[cs.x][i].x][0]));                }                else if(dis[cs.x][0]+v[cs.x][i].val<=dis[v[cs.x][i].x][1]){                    if(dis[cs.x][0]+v[cs.x][i].val==dis[v[cs.x][i].x][1]) cnt[v[cs.x][i].x][1]+=cnt[cs.x][0];                    else cnt[v[cs.x][i].x][1]=cnt[cs.x][0];                    dis[v[cs.x][i].x][1]=dis[cs.x][0]+v[cs.x][i].val;                    pq.push(node(v[cs.x][i].x,1,dis[v[cs.x][i].x][1]));                }            }            else{                if(dis[cs.x][1]+v[cs.x][i].val<=dis[v[cs.x][i].x][1]){                    if(dis[cs.x][1]+v[cs.x][i].val==dis[v[cs.x][i].x][1])cnt[v[cs.x][i].x][1]+=cnt[cs.x][1];                    else cnt[v[cs.x][i].x][1]=cnt[cs.x][1];                    dis[v[cs.x][i].x][1]=dis[cs.x][1]+v[cs.x][i].val;                    pq.push(node(v[cs.x][i].x,1,dis[v[cs.x][i].x][1]));                }            }        }    }}int main(){    int t;    cin>>t;    while(t--){        scanf("%d%d",&n,&m);        int a,b,c;        for(int i=0;i<=n;i++)            v[i].clear();        for(int i=0;i<m;i++){            scanf("%d%d%d",&a,&b,&c);            v[a].push_back(edge(b,c));                   }        scanf("%d%d",&x,&y);        dij(x,y);        if(dis[y][0]+1==dis[y][1])            cout<<cnt[y][1]+cnt[y][0]<<endl;        else            cout<<cnt[y][0]<<endl;    }    return 0;}