我大概是中毒了。。。。
对模拟退火 走火入魔了
这随机数算法。。。啧。。。
感觉有毒。
BZOJ2428: [HAOI2006]均分数据
【洛谷P2503】
已知N个正整数:A1、A2、……、An 。今要将它们分成M组,使得各组数据的数值和最平均,即各组的均方差最小。均方差公式如下:
,其中σ为均方差,是各组数据和的平均值,xi为第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
对于全部的数据,保证有K<=N <= 20,2<=K<=6
#include<cmath>
#include<cstdlib>
#include<cstdio>
#include<ctime>
#define llf double
using namespace std;
const int N=101;
const int TIME=5000;
const llf INF=2147483647;
int n,m,a[N];
llf ave,mn=INF;
llf pf(llf x){return x*x;}
void work(){
llf ans=0,T=10000;
int x[N]={0},bl[N]={0};
//随机分组
for(int i=1;i<=n;i++){
int grp=rand()%m+1;
bl[i]=grp;x[grp]+=a[i];
}for(int i=1;i<=m;i++)
ans+=pf(x[i]-ave);
//开始降温
while(T>0.1){
//随机出来一个需要调整的数字
int t=rand()%n+1,u=bl[t],v;
llf tmp=ans; T*=0.9;
if(T>500){
int grp;llf mnx=INF;
for(int i=1;i<=m;i++)
if(i!=u&&x[i]<mnx)
mnx=x[i],grp=i;
v=grp;
}else {
v=rand()%m+1;
while(u==v)v=rand()%m+1;
}
//把这个数字从这组转到另一组
ans-=pf(x[u]-ave)+pf(x[v]-ave);
x[u]-=a[t];x[v]+=a[t];
ans+=pf(x[u]-ave)+pf(x[v]-ave);
//答案合适 就看命修改
if(ans>tmp&&(rand()%10000>T))
ans=tmp,x[u]+=a[t],x[v]-=a[t];
else bl[t]=v;
}if(ans<mn)mn=ans;
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
ave+=a[i];
} ave/=m;
for(int i=1;i<=TIME;i++) work();
printf("%.2lf\n",sqrt(mn/m));
return 0;
}
好像也挺好玩的???23333.
貌似难在时间复杂度的把握上(⊙o⊙)