洛谷P8191 [USACO22FEB] Moo Network G
有 n n n个点,第 i i i个点的位置为 ( x i , y i ) (x_i,y_i) (xi,yi),其中 0 ≤ x i ≤ 1 0 6 , 0 ≤ y i ≤ 10 0\leq x_i\leq 10^6,0\leq y_i\leq 10 0≤xi≤106,0≤yi≤10。连接两个点 i , j i,j i,j的成本是 ( x i − x j ) 2 + ( y i − y j ) 2 (x_i-x_j)^2+(y_i-y_j)^2 (xi−xj)2+(yi−yj)2,求能使这 n n n个点连通的最小成本。
1 ≤ n ≤ 1 0 5 1\leq n\leq 10^5 1≤n≤105
观察数据范围发现, y y y的范围不超过 10 10 10,而连最短的边显然更优,所以对于每个点,我们可以将其与距离其较近的点连边,然后在这些边中做 kruskal \text{kruskal} kruskal即可。
先将点按横坐标从小到大排序,横坐标相同的按纵坐标排序,然后对于每个点,向左找 30 30 30个点连边,再在这些边中做 kruskal \text{kruskal} kruskal。因为 y y y的范围不超过 10 10 10,所以最小生成树中的边一般都在上面建的这些边中。
时间复杂度为 O ( 30 n log n ) O(30n\log n) O(30nlogn)。
#include
using namespace std;
int n,v1=0,fa[100005];
long long ans=0;
struct node{
int x,y;
}w[100005];
struct line{
int x,y;
long long dis;
}v[3000005];
bool cmp1(node ax,node bx){
if(ax.x!=bx.x) return ax.x<bx.x;
return ax.y<bx.y;
}
bool cmp2(line ax,line bx){
return ax.dis<bx.dis;
}
long long dis(int i,int j){
return 1ll*(w[i].x-w[j].x)*(w[i].x-w[j].x)+1ll*(w[i].y-w[j].y)*(w[i].y-w[j].y);
}
int find(int ff){
if(fa[ff]!=ff) fa[ff]=find(fa[ff]);
return fa[ff];
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d%d",&w[i].x,&w[i].y);
fa[i]=i;
}
sort(w+1,w+n+1,cmp1);
for(int i=1;i<=n;i++){
for(int j=max(i-30,1);j<i;j++){
v[++v1]=(line){i,j,dis(i,j)};
}
}
sort(v+1,v+v1+1,cmp2);
for(int i=1;i<=v1;i++){
int r1=find(v[i].x),r2=find(v[i].y);
if(r1!=r2){
fa[r1]=r2;
ans+=v[i].dis;
}
}
printf("%lld",ans);
return 0;
}