HDU-1233【最小生成樹+kruskal算法】

HDU-1233【最小生成樹+kruskal算法】

【模板水題】

#include <iostream>
#include <algorithm>
using namespace std;

static const int maxn = 100;

int p[maxn];

struct Edge
{
    int u, v, cost;
} roads[5000];  // 最多的路線就是5000條路線

int find(int x)
{
    return p[x] == x ? x : p[x] = find(p[x]);
}

int cmp(Edge x, Edge y)
{
    return x.cost < y.cost;
}

int main(int argc, const char **argv)
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0),cout.tie(0);
    
    int n; // n表示村莊數量
    int k;
    while (cin >> n && n)
    {
        k = n * (n - 1) / 2; // 表示路的條數
        for (int i = 0; i < k; i++)
        {
            cin >> roads[i].u >> roads[i].v >> roads[i].cost;
        }
        sort(roads, roads + k, cmp);
        for (int i = 0; i <= n; i++)
            // 初始化操作
            p[i] = i;
        int ans = 0, cnt = 0;
        for (int i = 0; i < k; i++)
        {
            int t1 = find(roads[i].u);
            int t2 = find(roads[i].v);
            // 當兩個結點不相等的時候
            if (t1 != t2)
            {
                ans += roads[i].cost;
                p[t1] = t2;
                cnt++;
            }
            if (cnt == n - 1)
                break;
        }
        cout << ans << endl;
    }
    return 0;
}