【题解】Luogu P2212 [USACO14MAR] 浇地 Watering the Fields 最小生成树

裸的板子，判一下d和c的大小

我菜死了，kruskal写挂的原因竟然是，sort的范围错了

WA到爆了

kruskal都忘了怎么写，赶快滚去复习

code

 

#include <bits/stdc++.h>
using namespace std;
namespace gengyf{
#define ll long long
const int maxn=1e6+10;
inline int read(){
    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;
}
int n,c,x[maxn],y[maxn],fa[maxn];
struct edge{
    int u,v;
    ll w;
}e[maxn*2];
int head[maxn],cnt;
inline void add(int u,int v,int w){
    e[++cnt].u=u;e[cnt].v=v;e[cnt].w=w;
}
bool cmp(edge a,edge b){
    return a.w<b.w;
}
int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
void kruskal(){
    int ans=0,tot=0;
    for(int i=1;i<=cnt;i++){
        int u=find(e[i].u),v=find(e[i].v);
        if(u!=v){
            fa[u]=v;
            ans+=e[i].w;tot++;
        }
        if(tot==n-1)break;
    }
    if(tot==n-1)printf("%d\n",ans);
    else puts("-1");
}
int main(){
    n=read();c=read();
    for(int i=1;i<=n;i++){
        x[i]=read();y[i]=read();
        for(int j=1;j<i;j++){
            int d=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
            if(d>=c){
                add(i,j,d);
            }
        }
    }
    sort(e+1,e+1+cnt,cmp);
    for(int i=1;i<=n;i++)fa[i]=i;
    kruskal();
    return 0;
}
}
signed main(){
  gengyf::main();
  return 0;
}