题目描述
某旋臂中存在着n颗恒星,求相距最近的两颗。
输入
第一个数为恒星的颗数n
接下来n行,每行3个浮点数x,y,z,表示恒星的坐标
输出
一个浮点数,表示相聚最近的两颗恒星的距离,保留4位小数。
输入样例
2
1.0 1.0 1.0
0.0 0.0 0.0
输出样例
1.7321
数据范围
n <= 1e6
思路分析
典型的最小点对问题,由二维改成了三维,基本思路没有太大变化。
1.排序:先按x排,x相同再按y排,y相同再按z排
(这个排序方式的变化,再qsort函数的cmp函数中修改)
2.使用FindMinLength函数分治:
点对只可能有三种位置方式:①都在左半侧②都在右半侧③跨中点分别在左右半侧
先递归分别求左半侧和右半侧的最小距离点对dis_left、dis_right,再求跨中点的点对中的最小距离对,最后取三者的最小值。
递归部分:
从原数组开始,分别求左右子数组
再对左右子数组进行同样操作
直到递归到子数组中只包含两个点时,返回两点间距离。子数组中只包含一个点时,返回最大值maxnum。
求跨中点最小距离点对的方法:
先取左右子数组最小距离的较小值,赋给dis_min,再用dis_min取缩小数组范围,最后在缩小的数组中挨个遍历,一方TLE。
缩小数组可以用每一个点与mid点的x距离与dis_min比较,如果某一个点与mid的x距离都比dis_min大,那么它不可能再与右半侧数组中的某一点组成距离比dis_min小的点对。
缩小数组范围后(temp[]储存缩小的数组的下标),再从头开始遍历所有两点间的距离寻找最小值,就不会出现TLE。
AC代码
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define maxnum 2147483646
struct node{
double x;
double y;
double z;
};
typedef struct node loc;
loc data[1000050];
int temp[1000050]={0};
double calculate(loc a,loc b);
int cmp(const void *a,const void *b);
double FindMinLength(int low,int high);
double min(double a,double b);
int main()
{
int i,n;
scanf("%d",&n);
for(i=0;i<n;i++)
scanf("%lf%lf%lf",&data[i].x,&data[i].y,&data[i].z);
qsort(data,n,sizeof(data[0]),cmp);
printf("%.4f",FindMinLength(0,n-1));
}
double calculate(loc a,loc b)
{
return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)+(a.z-b.z)*(a.z-b.z));
}
int cmp(const void *a,const void *b)
{
loc ac=*(loc*)a;
loc bd=*(loc*)b;
if(ac.x!=bd.x)
return ac.x-bd.x;
else if(ac.y!=bd.y)
return ac.y-bd.y;
else return ac.z-bd.z;
}
double min(double a,double b)
{return a<b?a:b;}
double FindMinLength(int low,int high)
{
double dis_min=maxnum;
int i,j,k=0;
if(low==high) return maxnum;
if(high-low==1) return calculate(data[low],data[high]);
int mid=(low+high)/2;
double dis_left,dis_right;
dis_left=FindMinLength(low,mid);
dis_right=FindMinLength(mid+1,high);
dis_min=min(dis_left,dis_right);
for(i=low;i<=high;i++)
{
if(fabs(data[mid].x-data[i].x)<=dis_min)
temp[k++]=i;
}
for(i=0;i<k;i++)
for(j=i+1;j<k;j++)
if(fabs(data[temp[j]].y-data[temp[i]].y)<dis_min&&fabs(data[temp[j]].z-data[temp[i]].z)<dis_min)
dis_min=min(dis_min,calculate(data[temp[i]],data[temp[j]]));
return dis_min;
}
总结
1.第一次本地运行不通过显示 error : SIGSEGV,找了很久的BUG,最后发现是temp[]数组在函数内部不能开这么大。
嗯,本地运行可以通过了。
2.提交到OJ上,wow,REG(SIGSEGV),还是数组内存有问题,又找了很久的BUG,发现是储存数组下标的变量k没有初始化。在本地系统给自动初始化为0了,没有出现问题,但在OJ上系统初始化成一个特别大的或负的数,就超出数组内存了。
3.改过来,再次兴冲冲地提交上去,WA(哇——)
又开始debug……
是因为一开始我把二维改成三维的时候,用
if(fabs(data[mid].x-data[i].x)<=dis_min&&fabs(data[mid].y-data[i].y)<=dis_min)
来缩小遍历范围,但实际上这个圈出来的范围偏小了,因为可能有的点离mid的y距离大于dis_min,但离其他点的距离小于dis_min。
所以只用x来缩小范围,在缩小的范围里遍历每两个点。
综上所述,我真是个智障.jpg。
PS: 听说这个题可以用奇怪的卡数据方法过去()