【ybtoj 3.2 最小生成树课堂过关】C.公路建设【kruskal优化】

题目链接：公路建设

分析

kruskal求最小生成树
然后算费用

上代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<algorithm>
using namespace std;

int n,m;
int fa[501];

struct lwx
{
	int x,y,z;
}a[2001];

int father(int x)
{
	if(fa[x]==x) return x;
	else return fa[x]=father(fa[x]);
 } 

int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
		int k=i-1;
		while(a[i].z<a[k].z&&k>=1) k--;
		k++;
		lwx t=a[i]; 
		for(int j=i;j>k;j--)
		{
			a[j]=a[j-1];
		}
		a[k]=t;
		for(int j=1;j<=n;j++)
		{
			fa[j]=j;
		}
		double ans=0;
		k=0;
		for(int j=1;j<=i;j++)
		{
			int fx=father(a[j].x);
			int fy=father(a[j].y);
			if(fx!=fy)
			{
				k++;
				fa[fx]=fy;
				ans+=(a[j].z*1.0)/2.0;
				if(k==n-1) break;
			}
		}
		if(k==n-1) cout<<setprecision(1)<<fixed<<ans<<endl;
		else cout<<0<<endl;
    }
	return 0;
}