[HAOI2006]均分数据(模拟退火????)

我大概是中毒了。。。。
对模拟退火 走火入魔了
这随机数算法。。。啧。。。
感觉有毒。

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
#include
#include
#include
#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]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(ansint 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⊙)

你可能感兴趣的:(c++深搜)