不知道比赛的时候是怎么想的,居然没有想到并查集,肯定是太久没做,失去对并查集的敏感度了。。。。
数据范围比较小,所有的边也只有100W而已,明显可以过的。
感觉就是最小生成树的思想,或者是一部分的最小生成树
简单并查集
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=1111;
int x[maxn],y[maxn];
struct in
{
int u,v,w;
}edge[maxn*maxn];
int belg[maxn];
int findx(int x)
{
return x==belg[x]?x:findx(belg[x]);
}
void Union(int a,int b)
{
int ra=findx(a),rb=findx(b);
belg[ra]=rb;
}
bool cmp(in x,in y)
{
return x.w<y.w;
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
if(n==0)
break;
for(int i=0;i<n;i++)
scanf("%d%d",&x[i],&y[i]);
int tot=0;
for(int i=0;i<n;i++)
{
belg[i]=i;
for(int j=i+1;j<n;j++)
{
edge[tot].u=i;
edge[tot].v=j;
edge[tot++].w=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
}
}
int s,e;
scanf("%d%d",&s,&e);
s-=1;
e-=1;
sort(edge,edge+tot,cmp);
int ans;
for(int i=0;i<n*n;i++)
{
if(findx(edge[i].u)!=findx(edge[i].v))
Union(edge[i].u,edge[i].v);
if(findx(s)==findx(e))
{
ans=edge[i].w;
break;
}
}
if(s==e)
ans=0;
printf("%d\n",ans);
}
return 0;
}