每日一题 - 240412 - 聪明的猴子

最新推荐文章于 2024-07-19 19:11:19 发布

乐意奥AI

最新推荐文章于 2024-07-19 19:11:19 发布

阅读量291

点赞数 9

分类专栏： # 蓝桥杯文章标签：算法

本文链接：https://blog.csdn.net/CodeToLiao/article/details/137681752

版权

蓝桥杯专栏收录该内容

25 篇文章 0 订阅

订阅专栏

本文介绍了Kruskal算法用于解决最小生成树问题，通过C++代码实现，涉及数据结构如并查集，以及时间复杂度为O(N*logN)。通过输入节点和边的信息，计算满足条件的连接数。

摘要由CSDN通过智能技术生成

聪明的猴子

TAG - $模板题、算法 - 【图论 - 最小生成树 - Kr u s ka l 】$
时间复杂度 - $\ast \log N)$

//
#include <bits/stdc++.h>
using namespace std;
// #define int long long 
#define data da

const int N = 2222;
int in[N], x[N], y[N], dad[N];
int n, m;
double maxx;

struct A {
    int x, y;
    double data;
    bool operator < (const A& in) const {
        return data < in.data;
    }
};
vector<A> e;

int get_dad(int x) {
    return dad[x] = dad[x] == x ? x : get_dad(dad[x]);
}

bool kruskal() {
    sort(e.begin(), e.end());               //
    for (int i = 1; i <= n; i++) dad[i] = i;

    int cnt = 0;
    for (auto [x, y, data] : e) {
        x = get_dad(x);
        y = get_dad(y);
        if (x != y) {
            dad[x] = y;
            maxx = max(maxx, data);
            cnt++;
        }
    }
    return cnt == n - 1;
}

void solve() {
    scanf("%d", &m);
    for (int i = 1; i <= m; i++) scanf("%d", &in[i]);

    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%d%d", &x[i], &y[i]);
    for (int i = 1; i <= n; i++) {
        for (int j = i + 1; j <= n; j++) {
            e.push_back((A){i, j, sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]))});
        }
    }

    kruskal();
    int ans = 0;
    for (int i = 1; i <= m; i++) {
        ans += (in[i] >= maxx);
    }
    printf("%d\n", ans);
}

signed main() {
    int t = 1;
    // scanf("%d", &t);
    while (t--) solve();
    return 0;
}