题目描述
爱好天文的小七发明了一种太空分区方法,在这个方法中,宇宙里亮度相近的星星被划为同一个星区。空间中任意相邻或坐标相同的星星若亮度差不大于给定整数M,则这两个星星属于同一星区(两个星星可能会有相同的坐标)。
现给你一个空间n个星星的三维坐标(x,y,z),和亮度值v,每个星星的上、下、左、右、前、后的六个相邻位置被认为是与其相邻的。请你计算一下该空间内的星区数量。
输入
第一行1个正整数n。n<=1000
第二行 1个正整数m。m<=109
第二行到第n+1行,每行四个正整数x,y,z,v。0<=x,y,z,v<=109
输出
一行正整数k,代表星区数量。
样例输入 Copy
5
2
1 1 1 2
1 2 3 2
1 2 2 5
1 1 0 3
1 0 1 3
样例输出 Copy
3
解析:
暴力1000*1000 并查集维护个数
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
ll m;
struct node
{
ll x,y,z,v;
}a[100005];
int fa[100005];
int find(int x)
{
if(x!=fa[x]) return fa[x]=find(fa[x]);
return fa[x];
}
bool check(node a,node b)
{
if(abs(a.v-b.v)>m) return false;
return abs(a.x-b.x)+abs(a.y-b.y)+abs(a.z-b.z)<=1; //判断是否相邻,相邻和重合都是<=1的,如果>1则是多个方向的
}
int main()
{
scanf("%d %lld",&n,&m);
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=n;i++)
{
cin>>a[i].x>>a[i].y>>a[i].z>>a[i].v;
}
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
{
if(check(a[i],a[j]))
{
int u=find(i);int v=find(j);
if(u!=v)
{
fa[u]=v;
}
}
}
int cnt=0;
for(int i=1;i<=n;i++)
{
if(i==fa[i]) cnt++;
}
cout<<cnt<<endl;
}