P1856 Roads in Berland C++详细代码

这是一道经典的最小生成树问题，可以使用 Prim 或者 Kruskal 算法解决。下面给出一个使用 Prim 算法的详细代码：

#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;

const int N = 1005;
int n, m;
int g[N][N], d[N];
bool st[N];

int prim()
{
    memset(d, 0x3f, sizeof d);
    int res = 0;
    for (int i = 0; i < n; i ++ )
    {
        int t = -1;
        for (int j = 1; j <= n; j ++ )
            if (!st[j] && (t == -1 || d[t] > d[j]))
                t = j;
        if (i && d[t] == 0x3f3f3f3f) return -1;
        if (i) res += d[t];
        st[t] = true;
        for (int j = 1; j <= n; j ++ )
            d[j] = min(d[j], g[t][j]);
    }
    return res;
}

int main()
{
    cin >> n >> m;
    memset(g, 0x3f, sizeof g);
    while (m -- )
    {
        int a, b, c;
        cin >> a >> b >> c;
        g[a][b] = g[b][a] = min(g[a][b], c);
    }
    int t = prim();
    if (t == -1) puts("impossible");
    else cout << t << endl;
    return 0;
}

时间复杂度为 $O(n^2)$，可以通过本题。