P2212 浇地

题面:https://www.luogu.org/problemnew/show/P2212

本题可以求出n个点的两两间距离,然后建图跑最小生成树即可。

Code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<ctime>
using namespace std;
const int N=1000005;
int n,m,u1,v1,Cnt,fa[N],ans,x[N],y[N],c,cou;
struct Node{
    int u,v,w;
}Edge[N*2];
bool cmp(Node a,Node b){
    return a.w<b.w;
}
int find(int x){
    if(fa[x]!=x){
        fa[x]=find(fa[x]);
    }
    return fa[x];
}
void kruskal(){
    sort(Edge+1,Edge+1+cou,cmp);
    for(int i=1;i<=cou;i++){
        u1=find(Edge[i].u);
        v1=find(Edge[i].v);
        if(u1==v1){
            continue;
        }
        ans+=Edge[i].w;
        fa[v1]=u1;
        Cnt++;
        if(Cnt==n-1){
            break;
        }
    }
}
int main(){
    scanf("%d%d",&n,&c);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&x[i],&y[i]);
        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){
                Edge[++cou].u=i;
                Edge[cou].v=j;
                Edge[cou].w=d;
            }
        }
    }
    for(int i=1;i<=n;i++){
        fa[i]=i;
    }
    kruskal();
    if(Cnt==n-1){
        printf("%d\n",ans);
    }
    else{
        printf("-1\n");
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ukcxrtjr/p/11184683.html
0条评论
添加一条新回复