黑暗城堡（最小路径生成树）

原题链接

解题思路：最短路径生成树方案数，第一步想用dijkstra算出每个点到1号点的最短距离，然后再用prim算法判断每个未加入到树中的点可以由几个已经加入到树中的点拓展。将结果累乘即为最终的答案

C++代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 1010;
const int mod = (1 << 31) - 1;
int n, m;
int w[N][N];
int d[N];
bool st[N];

void dijkstra() //dijkstra求所有点到1号点的最短距离
{
    memset(d, 0x3f, sizeof d); //初始化距离
    d[1] = 0; //起点距离为0
    memset(st, 0, sizeof st); //在本函数中表示判重数组，初始化判重数组
    for(int i = 0; i < n - 1; i++) //朴素dijkstra
    {
        int t = -1;
        for(int j = 1; j <= n; j++)
            if(!st[j] && (t == -1 || d[t] > d[j]))
                t = j;
        st[t] = true;
        for(int j = 1; j <= n; j++)
            d[j] = min(d[j], d[t] + w[t][j]);
    }
}

int prim() //求最短路径生成树的方案数
{
    memset(st, 0, sizeof st); //在本函数中表示每个点是否在最短路径生成树中
    st[1] = true; //1号点最开始就在最短路径生成树中
    int res = 1; //记录最短路径生成树的总方案数，1号点是第一个点，只有一种方案
    for(int i = 0; i < n - 1; i++) //计算剩下 n - 1 个点的方案数
    {
        int t = -1;
        for(int j = 2; j <= n; j++)
            if(!st[j] && (t == -1 || d[t] > d[j]))
                t = j;

        LL cnt = 0; //记录当前点的方案数
        for(int j = 1; j <= n; j++) //枚举t能从哪个点加入最短路径生成树
            if(st[j] && d[t] == d[j] + w[j][t]) //如果t从当前点加入满足最短路径生成树的条件
                cnt++; //方案数+1

        st[t] = true; //标记t已加入最短路径生成树
        res = (LL)res * cnt % mod; //累乘方案数
    }
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    memset(w, 0x3f, sizeof w); //初始化邻接矩阵
    while(m--)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        w[a][b] = w[b][a] = min(w[a][b], c); //无向边
    }
    dijkstra(); //dijkstra求所有点到1号点的最短距离
    printf("%d\n", prim()); //求最短路径生成树的方案数
    return 0;
}

参考博客