qzezoj 1641 黑暗城堡

题面传送门
题目中要求最小路径生成树的方案数。
先跑一遍\(spfa\)把最小路径跑出来，然后对于每个点枚举边，看看有几个点是可以转移，然后乘法原理计数即可。
代码实现:

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define mod 2147483647
using namespace std;
int n,m,k,x,y,z,d[1039],now,cur;
long long ans=1,tot;
struct yyy{int to,w,z;}tmp;
struct ljb{
    int head,h[1039];
    yyy f[1000039];
    inline void add(int x,int y,int z){
        f[++head]=(yyy){y,z,h[x]};
        h[x]=head;
    }
}s;
queue<int > q;
inline void read(int &x){
    char s=getchar();x=0;
    while(s<'0'||s>'9') s=getchar();
    while(s>='0'&&s<='9') x=(x<<3)+(x<<1)+(s^48),s=getchar();
}
int main(){
    register int i;
    memset(s.h,-1,sizeof(s.h));
    memset(d,0x3f,sizeof(d));
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++)read(x),read(y),read(z),s.add(x,y,z),s.add(y,x,z);
    q.push(1);
    d[1]=0;
    while(!q.empty()){
        now=q.front();
        q.pop();
        cur=s.h[now];
        while(cur!=-1){
            tmp=s.f[cur];
            if(d[tmp.to]>d[now]+tmp.w) d[tmp.to]=d[now]+tmp.w,q.push(tmp.to);
            cur=tmp.z;
        }
    }
    for(i=2;i<=n;i++){
        //printf("%d\n",d[i]);
        tot=0;
        cur=s.h[i];
        while(cur!=-1){
            tmp=s.f[cur];
            if(d[tmp.to]+tmp.w==d[i]) tot++;
            cur=tmp.z;
        }
        ans=ans*tot%mod;
    }
    printf("%lld\n",ans);
}