BZOJ2428 均分数据

2428: [HAOI2006]均分数据

Time Limit: 5 Sec  Memory Limit: 128 MB

Description

已知N个正整数：A1、A2、……、An 。今要将它们分成M组，使得各组数据的数值和最平均，即各组的均方差最小。均方差公式如下：

,其中σ为均方差，是各组数据和的平均值，x_i为第i组数据的数值和。

Input

第一行是两个整数，表示N,M的值（N是整数个数，M是要分成的组数）

第二行有N个整数，表示A1、A2、……、An。整数的范围是1--50。

（同一行的整数间用空格分开）

Output

这一行只包含一个数，表示最小均方差的值(保留小数点后两位数字)。

Sample Input

6 3
1 2 3 4 5 6

Sample Output

0.00

HINT

对于全部的数据，保证有K<=N <= 20，2<=K<=6

模拟退火，随机选点更换分组

当温度太大不稳定，直接尝试换到sum最小的组

由于模拟退火的不稳定，所以跑个几万遍就稳了233

当然如果您是yzh那样的强者，您可以用DP每次算最优解

个人认为这样并不优秀，如果您有心情$n ^ 2 m$DP，为何不多退$n ^ 2 m$次火<_<

#include<bits/stdc++.h>
using namespace std;
template <class _T> inline void read(_T &_x) {
	int _t; bool flag = false;
	while ((_t = getchar()) != ‘-‘ && (_t < ‘0‘ || _t > ‘9‘)) ;
	if (_t == ‘-‘) _t = getchar(), flag = true; _x = _t - ‘0‘;
	while ((_t = getchar()) >= ‘0‘ && _t <= ‘9‘) _x = _x * 10 + _t - ‘0‘;
	if (flag) _x = -_x;
}
using namespace std;
const int maxn = 30;
int n, m, a[maxn];
double ave;
int from[maxn], sum[maxn];
inline double sqr(double val) {return val * val; }
inline double getr() {return (double)rand() / RAND_MAX; }
inline double solve() {
	memset(sum, 0, sizeof sum);
	for (int i = 1; i <= n; ++i) {
		from[i] = rand() % m + 1;
		sum[from[i]] += a[i];
	}
	double ans = 0;
	for (int i = 1; i <= m; ++i)
		ans += sqr(sum[i] - ave);
	double T = 10000;
	while (T > 0.1) {
		int t = rand() % n + 1, x = from[t], y;
		if (T > 1000) y = min_element(sum + 1, sum + m + 1) - sum;
		else y = rand() % m + 1;
		double to = ans;
		to -= sqr(sum[x] - ave) + sqr(sum[y] - ave);
		to += sqr(sum[x] - a[t] - ave) + sqr(sum[y] + a[t] - ave);
		if (to < ans || exp((ans - to) * 1e4 / T) > getr()) {
			ans = to;
			from[t] = y;
			sum[x] -= a[t], sum[y] += a[t];
		}
		T *= 0.9;
	}
	return ans;
}
int main() {
	//freopen();
	//freopen();
	srand(19260817);
	read(n), read(m);
	for (int i = 1; i <= n; ++i) {
		read(a[i]);
		ave += a[i];
	}
	ave /= m;
	double ans = solve();
	for (int i = 1; i <= 10000; ++i)
		ans = min(ans, solve());
	printf("%.2lf\n", sqrt(ans / m));
	return 0;
}