[WOJ1182]最小方差生成树

传送门

我们考虑枚举所有边的和s, 除以n-1就是平均数, 边权重新定义为(ai - average) ^ 2

然后跑最小生成树, 如果老的边权和是s, 说明可以作为答案

我们发现答案一定可以在某个s中被枚举到, 不用担心漏解的问题

还可以加一个可行背包的优化, 如果加起来都达不到这个数就不用枚举了

#include<bits/stdc++.h>
#define N 105
#define M 2005
#define K 10005
#define inf 0x3fffffff
using namespace std;
int read(){
	int cnt=0,f=1; char ch=0;
	while(!isdigit(ch)){ch=getchar(); if(ch=='-') f=-1;}
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt * f;
}
int n,m,Min,Max,f[K]; double Ans = inf;
struct Node{
	int u,v,c; double tmp;
}E[M];
bool cmp (Node a,Node b){ return a.tmp < b.tmp;}
int fa[N]; int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void Check(int A){
	for(int i=1;i<=n;i++) fa[i] = i;
	double s = A*1.0 / (n-1)*1.0;
	for(int i=1;i<=m;i++){
		E[i].tmp = (E[i].c * 1.0 - s) * (E[i].c * 1.0 - s);
	} sort(E+1,E+m+1,cmp);
	
	int cnt=0, sum=0; double ans = 0;
	for(int i=1;i<=m,cnt<n-1;i++){
		int x = E[i].u, y = E[i].v;
		int fx = find(x), fy = find(y);
		if(fx!=fy){
			fa[fx] = fy;
			cnt++; sum += E[i].c; ans += E[i].tmp;
		}
	} 
	if(sum == A) Ans = min(Ans, ans);
}
int main(){
	n = read(), m = read(); f[0] = 1;
	for(int i=1;i<=m;i++){
		E[i] = (Node){read(), read(), read()};
		E[i].tmp = E[i].c;
		for(int j=K-5;j>=E[i].c;j--) f[j] |= f[j-E[i].c];
	} sort(E+1, E+m+1, cmp);
	for(int i=1;i<=n-1;i++) Min += E[i].c;
	for(int i=m-n+1;i<=m;i++) Max += E[i].c;
	for(int i=Min;i<=Max;i++) if(f[i]) Check(i);
	Ans /= (n-1) * 1.0; Ans = sqrt(Ans);
	printf("%0.4lf",Ans); return 0;
}