loj#10064 黑暗城堡

 

求图中的最短路径生成树有多少个？ （ 该生成树中的任意点i ，i到1的距离和  原图中的i到1的最短距离相等

 

 跑所有点到1的单源最短路，d[i]

 if d[i]==d[y]+z ，那么z这个路径是该生成树的一条边

然后就是多步计数，乘法原理即可

 

#include <iostream>
#include <cstring>
#include <queue>
using namespace std ; 
 #define int long long
 struct T{
     int y,z;
     T(int y0,int z0){
         y=y0,z=z0;
     }
     bool operator<(T a)const{
         return z>a.z;
     }
 };
 const int N=1e4+1,M=1e6+1,inf=0x3f3f3f3f;
 const int mod=(1<<31)-1;
 int n,b[N],d[N],nxt[M],w[M],go[M],hd[N],all;
 int m;
 int cnt[N];
 void add(int x,int y,int z){
     go[++all]=y,nxt[all]=hd[x]; hd[x]=all,w[all]=z;
 }
 priority_queue<T> q;
 
 void dijk(int v0){
     int i,x,y,z;
     memset(d,inf,sizeof d); d[v0]=0;
     q.push(T(v0,0));
     
     while(q.empty()==0){
         x=q.top().y,q.pop(); if(b[x]) continue;
         b[x]=1;
         for(i=hd[x];i;i=nxt[i]){
             y=go[i],z=w[i]; 
             if(d[x]+z<d[y]){
                 d[y]=d[x]+z; q.push(T(y,d[y]));
            }
         }
     }
 }
 void solve(){
     int i,j,x,y,z;
     for(i=1;i<=m;i++) cin>>x>>y>>z,add(x,y,z),add(y,x,z);
     
     dijk(1);
     for(i=1;i<=n;i++)
      for(j=hd[i];j;j=nxt[j]){
          y=go[j]; if(d[y]==d[i]+w[j]) cnt[y]++;
      }
      int ans=1;
      for(i=2;i<=n;i++) ans*=cnt[i],ans%=mod;
      cout<<ans;
 }
 signed main(){
     cin>>n>>m;
     solve();
 }