Libre OJ #10064 黑暗城堡（spfa+STL求短路）

题目链接：https://loj.ac/problem/10064


分析：
若x是y的父节点，x，y之间的通道长度为z，则应该有dist[y]=dist[x]+z;
spfa求单源最短路径。利用vector更为简便。
代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
typedef long long LL;
const int N=1e3+10;
const int INF=0x3f3f3f3f;
const int mod=(1<<31)-1;
int n,m,tot;
int book[N],dis[N],cnt[N];
struct node
{
    int to,ds;
};
vector<node>e[N];
void spfa()
{
    memset(dis,INF,sizeof(dis));
    queue<int>q;
    book[1]=1;
    dis[1]=0;
    q.push(1);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        book[u]=0;/*出对列*/
        for(int i=0; i<e[u].size(); i++)/*遍历u的所有出边*/
        {
            int v=e[u][i].to,w=e[u][i].ds;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                if(book[v]==0)
                {
                    book[v]=1;
                    q.push(v);
                }
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    while(m--)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        e[u].push_back({v,w});/*存图*/
        e[v].push_back({u,w});
    }
    spfa();
    memset(cnt,0,sizeof(cnt));
    for(int i=1; i<=n; i++)
        for(int j=0; j<e[i].size(); j++)
        {
            int v=e[i][j].to;
            if(dis[i]+e[i][j].ds==dis[v])
                cnt[v]++;
        }
    LL ans=1;
    for(int i=2; i<=n; i++)
    {
        ans*=(LL)cnt[i];
        ans%=mod;
    }
    printf("%lld\n",ans);
    return 0;
}