Highways——最小生成树模板题(求距离)
作者:互联网
题意:
二维坐标系上, 给出n个顶点的坐标, 有m个组顶点之间距离为0, 输出最小生成树中所有距离不为0的边的两点的坐标
题解:
裸最小生成树板子
只需要求出任意两点之间的距离即可
代码:
#include<iostream> #include<stdio.h> #include<math.h> #include<algorithm> #include<vector> using namespace std; typedef long long ll; const int maxn = 750+10; const int maxm = 5e5; int f[maxn]; int x[maxn],y[maxn]; int n,cnt,m; struct node { int u,v; double w; bool operator < (const node &a)const { return w<a.w; } } edge[maxm]; int Find(int x) { return x==f[x]?x:f[x]=Find(f[x]); } void add(int u,int v,double w) { edge[cnt].u=u; edge[cnt].v=v; edge[cnt++].w=w; } void kruskal() { int sum=0; for(int i=0; i<=n; i++)f[i]=i; sort(edge,edge+cnt); for(int i=0; i<cnt; i++) { int x=edge[i].u; int y=edge[i].v; int fx=Find(x); int fy=Find(y); if(fx!=fy) { f[fx]=fy; sum++; if(edge[i].w>0)printf("%d %d\n",x,y); } if(sum==n-1)break; } } double get_dis(int i,int j) { double dis=sqrt((x[i]-x[j])*(x[i]-x[j])*1.0+(y[i]-y[j])*(y[i]-y[j])*1.0); //printf("%.2f\n",dis); return dis; } int main() { scanf("%d",&n); cnt=0; for(int i=0; i<=n; i++)f[i]=i; for(int i=1; i<=n; i++)scanf("%d%d",&x[i],&y[i]); for(int i=1; i<=n; i++)for(int j=i+1; j<=n; j++) { double dis=get_dis(i,j); add(i,j,dis); } scanf("%d",&m); while(m--) { int a,b; scanf("%d%d",&a,&b); add(a,b,0); } kruskal(); return 0; }kruskal
标签:const,int,double,最小,Highways,maxn,include,模板,dis 来源: https://www.cnblogs.com/j666/p/11616979.html