[BZOJ3754][最小方差生成树]tree

BZOJ3754

分析：
我们可以枚举一些实数，比如可以去枚举平均数。
设当前枚举到 x，这时建一个新图，原来边权为 w 的边权是 $(w-x{)}^2$，在这个新图上求最小生成树，设得到的权和是f[x]。
经研究发现有以下性质：
f[x]是一个分段函数，每一段都是一个下凸的二次函数。
f[x]上每一段的最低点都对应着一个合法的平均数，即，找得到一颗生成树恰是这个平均数。
所以，取f[x]的最小值就是答案。
设原图最小生成树权和为 L，最大生成树权和为 R ，那么平均数所在区间即 $[\frac{L}{n-1},\frac{R}{n-1}]$，且相邻平均数差最小都是$\frac{1}{n-1}$

Code：

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	char ch=getchar();
	int res=0,f=1;
	while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
	while(isdigit(ch))res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
	return res*f;
}
const int N=105;
const int M=2005;
int n,m,fa[N],mn=1e9,mx=0;
struct edge{
	int u,v,w,vis;
	double val;
}e[M];
double ans=1e9;
inline bool operator <(const edge &a,const edge &b){
	return a.val<b.val;
}
inline int find(int x){
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
inline double pf(double x){
	return x*x;
}
inline double Kruscal(){
	double res=0;
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++){
		int f1=find(e[i].u),f2=find(e[i].v);
		if(f1!=f2){
			fa[f1]=f2,res+=e[i].val;
		}
	}
	return res;
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=m;i++){
		e[i].u=read(),e[i].v=read(),e[i].val=e[i].w=read();
	}
	sort(e+1,e+m+1);
	mn=Kruscal();
	reverse(e+1,e+m+1);
	mx=Kruscal();
	for(int i=mn;i<=mx;i++){
		double ave=i*1.0/(double)(n-1);
		for(int j=1;j<=m;j++)e[j].val=pf((double)e[j].w-ave);
		sort(e+1,e+m+1);
		ans=min(ans,Kruscal());//cout<<ans<<'\n';
	}
	printf("%.4lf",(sqrt(ans/(double)(n-1))));
}