POJ 2421 Constructing Roads

Tony L.Du

于 2016-01-24 09:48:35 发布

阅读量375

点赞数

CC 4.0 BY-SA版权

分类专栏： POJ 最小生成树文章标签： poj

本文链接：https://blog.youkuaiyun.com/Dextrad_ihacker/article/details/50572842

POJ 同时被 2 个专栏收录

19 篇文章

订阅专栏

最小生成树

7 篇文章

订阅专栏

本文介绍了一种基于Kruskal算法实现最小生成树的具体过程。通过输入矩阵来构造图，并利用排序和并查集确定最小生成树的边，最终输出最小生成树的总权重。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：
[kuangbin带你飞]专题六最小生成树 D - Constructing Roads

输入矩阵，求最小生成树。其中有的边已经联通，不需要再计算，将其权值置为0即可。

//#include"stdafx.h"
#include <iostream>
#include <cstdio>
#include <cmath>
#include<cstring>
#include <algorithm>
using namespace std;

struct edge
{
    int s, e, w;
};

edge e[10008];
int p[108];
int n, ans = 0, cnt = 0;
int map[108][108];

bool cmp(edge a, edge b) { return a.w < b.w; }
int find(int x){ return x == p[x] ? x : p[x] = find(p[x]); }

void kruscal()
{
    sort(e, e + cnt, cmp);

    for (int i = 0; i < cnt; i++)
    {
        int x = find(e[i].s);
        int y = find(e[i].e);
        if (x != y)
        {
            p[x] = y;
            ans += e[i].w;
        }
    }
    cout << ans << endl;
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++) cin >> map[i][j];

    int m, x, y;
    for (int i = 0; i <= n; i++) p[i] = i;

    cin >> m;
    for (int i = 0; i < m; i++)
    {
        cin >> x >> y;
        //ans += map[x - 1][y - 1];
        map[x - 1][y - 1] = 0;
    }



    for (int i = 0; i < n - 1; i++)
        for (int j = i + 1; j < n; j++)
            {
                e[cnt].s = i;
                e[cnt].e = j;
                e[cnt++].w = map[i][j];
            }

    kruscal();

    return 0;

}