poj 3463 Sightseeing——次短路计数

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

当然要给一个点记最短路和次短路的长度和方案。

但往优先队列里放的结构体和vis竟然也要区分0/1，就像把一个点拆成两个点了一样。

不要区分k的fx。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=,M=;
int T,n,m,head[N],xnt,dis[N][],f[N][],st,en;
bool vis[N][];
struct Ed{
  int next,to,w;
  Ed(int n=,int t=,int z=):next(n),to(t),w(z) {}
}ed[M];
struct Node{
  int bh,dis;bool fx;
  Node(int b=,int d=,bool f=):bh(b),dis(d),fx(f) {}
  bool operator< (const Node &b)const
  {
    return dis>b.dis;
  }
};
void add(int x,int y,int z)
{
  ed[++xnt]=Ed(head[x],y,z);head[x]=xnt;
}
void dj()
{
  memset(vis,,sizeof vis);memset(dis,,sizeof dis);
  memset(f,,sizeof f);
  priority_queue<Node> q;
  dis[st][]=;//dis[st][1]
  f[st][]=;//!
  q.push(Node(st,,));//no push [1]
  while(q.size())
    {
      Node t=q.top();q.pop();int k=t.bh;bool fx=t.fx;
      while(q.size()&&vis[k][fx])t=q.top(),q.pop(),k=t.bh,fx=t.fx;
      if(vis[k][fx])break;vis[k][fx]=;
      for(int i=head[k],v;i;i=ed[i].next)//不要区分fx，[k][1]也能更新[v][0]！
    {
        int d=dis[k][fx],w=ed[i].w;
          if(d+w<dis[v=ed[i].to][])
        {
          dis[v][]=dis[v][];f[v][]=f[v][];
          dis[v][]=d+w;f[v][]=f[k][fx];
          q.push(Node(v,dis[v][],));
          q.push(Node(v,dis[v][],));//!because zj's [1] also has been pshp
        }
          else if(d+w==dis[v][])
        f[v][]+=f[k][fx];
          else if(d+w<dis[v][])
        {
          dis[v][]=d+w;f[v][]=f[k][fx];
          q.push(Node(v,dis[v][],));
        }
          else if(d+w==dis[v][])
        f[v][]+=f[k][fx];
    }
    }
}
int main()
{
  scanf("%d",&T);
  while(T--)
    {
      scanf("%d%d",&n,&m);
      memset(head,,sizeof head);xnt=;
      int x,y,z;
      for(int i=;i<=m;i++)
    {
      scanf("%d%d%d",&x,&y,&z);add(x,y,z);
    }
      scanf("%d%d",&st,&en);
      dj();
      if(dis[en][]==dis[en][]+)f[en][]+=f[en][];
      printf("%d\n",f[en][]);
    }
  return ;
}