1.数据可以直接用了,不需要再开平方了
2.走0捷径也好,常规求距离也罢,直接加边就可以了,毕竟sort可以排序不虚
3.只要在kruskal每次并查集过程中输入加入edge即可。
4.我在写Point 的时候直接在里面写length函数的时候卡壳了一下。。。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=750+10;
const int maxm=500000;
struct Edge
{
int u,v,dist;
Edge(){}
Edge(int u,int v,int d):u(u),v(v),dist(d){}
bool operator<(const Edge &rhs)const
{
return dist < rhs.dist;
}
};
struct Kruskal
{
int n,m;
Edge edges[maxm];
int fa[maxn];
int findset(int x){ return fa[x]==-1? x:fa[x]=findset(fa[x]); }
void init(int n)
{
this->n=n;
m=0;
memset(fa,-1,sizeof(fa));
}
void AddEdge(int u,int v,int dist)
{
edges[m++]=Edge(u,v,dist);
}
int kruskal()
{
int sum=0;
int cnt=0;
sort(edges,edges+m);
for(int i=0;i<m;i++)
{
int u=edges[i].u, v=edges[i].v;
if(findset(u) != findset(v))
{
if(edges[i].dist>0) printf("%d %d\n",u,v);
sum +=edges[i].dist;
fa[findset(u)] = findset(v);
if(++cnt>=n-1) break;
}
}
if(cnt<n-1) return -1;
return sum;
}
}KK;
struct Point
{
int x,y;
Point(){}
Point(int x0,int y0):x(x0),y(y0){}
}p[maxn];
int length(Point a,Point b){
return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);
}
int main()
{
int n;
scanf("%d",&n);
KK.init(n);
for(int i=1;i<=n;i++)
scanf("%d%d",&p[i].x,&p[i].y);
int t;
scanf("%d",&t);
int u,v;
for(int i=0;i<t;i++){
scanf("%d%d",&u,&v);
KK.AddEdge(u,v,0);
}
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++){
KK.AddEdge(i,j,length(p[i],p[j]));
}
KK.kruskal();
return 0;
}