题解 P2504 【[HAOI2006]聪明的猴子】

这道题要坑死人啊。。。

第一次做40分，其他RE，改了一次之后就变成20分了。。。

究其原因，是有一个数组（dis）开的太小了，于是最后本蒟蒻就随手开了个五百万，然后，就AC了。

看到有一篇题解说求距离最好不开根，经本蒟蒻实验，开不开根并不影响结果，全看个人喜好。

本蒟蒻还加了一个快读，其实也没有必要，只是第一次RE的时候不小心看成了TLE，为了缩短时间才写的。

总体来说，这是一道Kruskal的题废话。

需要注意的一个点在于题目所求的与裸题有差异。

其余的好像问题也不大，只要不像本蒟蒻一样视力不好就不会错得太离谱。

总而言之，AC代码如下（并查集find函数带优化）：

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

inline int qread() {
    int x = 0, f = 1;
    char c = getchar();
    while(c < '0' || c > '9') {
        if(c == '-') f = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9') {
        x = x * 10 + (c - '0');
        c = getchar();
    }
    return x * f;
}

const int maxn = 1010;

int m, n, ans = 0, cnt = 0;
int f[maxn], leap[maxn], x[maxn], y[maxn];

struct node {
    int x, y;
    int val;
}dis[5000005];

bool cmp(node a, node b) {
    return a.val < b.val;
}

int find(int x) {
    int r = x;
    while(r != f[r]) r = f[r];
    int i = x, j;
    while(f[i] != r) {  
        j = f[i];  
        f[i] = r;  
        i = j;  
    }
    return r;  
}

void merge(int x, int y) {
    x = find(x);
    y = find(y);
    if(x != y) f[y] = x;
}

double dt(int x1,int x2,int y1,int y2) {
    return sqrt(pow(double(x1 - x2), 2) + pow(double(y1 - y2), 2));
}

int main() {
    cin >> m;
    for(int i = 1; i <= m; i++) cin >> leap[i];
    cin >> n;
    for(int i = 1; i <= n; i++) cin >> x[i] >> y[i];
    for(int i = 1; i <= maxn - 1; i++) f[i] = i;
    
    for(int i = 1; i <= n; i++) 
        for(int j = i + 1; j <= n; j++) {
            dis[++cnt].x = i;
            dis[cnt].y = j;
            dis[cnt].val = dt(x[i], x[j], y[i], y[j]);
        }
            
            
    sort(dis + 1, dis + cnt + 1, cmp);
    for(int i = 1; i <= cnt; i++)
        if(find(dis[i].x) != find(dis[i].y)) {
            ans = dis[i].val;
            merge(dis[i].x, dis[i].y);
        }
        
    int sum = 0;
    for(int i = 1; i <= m; i++) if(leap[i] >= ans) sum++;
    
    cout << sum;
}