洛谷 P2872 [USACO07DEC]Building Roads S(最小生成树)

菜鸟生成记(60)

这一题就是最小生成树绕了一个小弯,

模板题：直接给出结点数和边数;以及边的两个顶点和边权;

这一题：给出n个坐标(x,y),这n个坐标就代表n个结点;通过两点间的距离公式sqrt(pow(x1-x2,2)+pow(y1-y2,2))求出每个点其他结点的距离(边权);然后按边权的升序把这些边排序;然后选取n-1个边构成最小生成树

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
using namespace std;
const int N=1e6+10;
/*
n<=1000;
n^2==1e6
数组开小了就RE了 
*/
int pre[N]={0};//记录祖辈数组 
struct st{
	int i,j;//坐标(s[]中)||两个结点编号(t[]中) 
	double w;//边权 
	bool operator<(const st &a)const
	{//排序规则,按边权升序 
		return w<a.w||false;
	}
}; 
int find1(int x)//并查集(查询祖辈函数) 
{
	int t=x;
	while(pre[t]!=t)
	{
		t=pre[t];
	}
	return t;
}
int main()
{
	st s[N];//储存坐标 
	st t[N];//储存结点和边权 
	int n,m1,k=0,num=0;
	double sum=0;
	cin>>n>>m1;
	for(int i=1;i<=n;i++)
	pre[i]=i;//初始化祖辈数组 
	for(int i=1;i<=n;i++)
	{
		cin>>s[i].i>>s[i].j;
	}
	for(int i=1;i<=n-1;i++)//双循环生成边结点和边权 
	{//这里生成边数接近于n^2(n<=1000)==1e6 
	//所以t[]和s[]数组大小,最好大于等于1e6 
		sum=0;
		for(int j=i+1;j<=n;j++)
		{
			int x,y,y1,x1;
			x=s[i].i;
			y=s[i].j;
			x1=s[j].i;
			y1=s[j].j;
			sum=sqrt(pow((double)(x-x1),2)+pow((double)(y-y1),2));
			t[k].i=i;//储存成边结点和边权 
			t[k].j=j;
			t[k++].w=sum;
		}
	}
	sort(t,t+k);//排序 
	for(int i=0;i<m1;i++)
	{
		int t1,t2,u,v;
		scanf("%d%d",&u,&v);//题中给出的边结点
		//所以u和v已经连接 
		t1=find1(u);
		t2=find1(v);
		if(t1!=t2)
		{//并查集(合并集合,结点并入生成树) 
			pre[t1]=t2;
			num++;//生成树边的计数 
		}
	}
	sum=0;
	for(int i=0;i<k;i++)
	{
		int t1,t2,u,v;
		u=t[i].i;
		v=t[i].j;
		t1=find1(u);
		t2=find1(v);
		if(t1!=t2)
		{//并查集(合并集合,结点并入生成树) 
			pre[t1]=t2;
			sum+=t[i].w;//代价累加 
			num++;//生成树边的计数 
		}
		if(num==n-1)//最小生成树生成 
		break;
	}
	printf("%.2lf\n",sum);//输出最小生成树代价(题中已连接的边权未加入其中) 
	return 0;
}