BZOJ.2428.[HAOI2006]均分數據(隨機化貪心/模擬退火)
阿新 • • 發佈:2018-06-12
BE -s ref source eps digi 很難 有時 define
題目鏈接
模擬退火:
模擬退火!每次隨機一個位置加給sum[]最小的組。
參數真特麽玄學啊。。氣的不想調了(其實就是想刷刷最優解)
如果用DP去算好像更準。。
//832kb 428ms
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <assert.h>
#include <algorithm>
#define gc() getchar()
#define D (0.997)//big enough!
#define eps (1e-3)
#define Rand(x) (rand()%x+1)
const int N=22;
int n,K,A[N],sum[7],bel[N];
double Ans=1e14,Aver,Tmp;
inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
inline double Squ(double x){
return x*x;
}
inline void Move(int fr,int to,int tar){
sum[fr]-=A[tar], sum[bel[tar]=to]+=A[tar];
}
//inline void Update(double &x){
// x=0.0; for(int i=1; i<=K; ++i) x+=Squ(sum[i]);
//}
void SA()
{
memset(sum,0,sizeof sum);
for(int i=1; i<=n; ++i) sum[bel[i]=Rand(K)]+=A[i];
double ans=0,nxt;
for(int i=1; i<=K; ++i) ans+=Squ(sum[i]);
Ans=std::min(Ans,ans+Tmp);
for(double T=1e7/*big enough*/; T>eps; T*=D)//Times:6898
{
int tar=Rand(n), fr=bel[tar], to=std::min_element(sum+1,sum+1+K)-sum;
if(fr==to) continue;
nxt=ans-Squ(sum[fr])-Squ(sum[to]);
Move(fr,to,tar);
nxt+=Squ(sum[fr])+Squ(sum[to]);
if(nxt<ans || (exp((ans-nxt)/T)*RAND_MAX>rand())) ans=nxt;//!
else Move(to,fr,tar);
Ans=std::min(Ans,ans+Tmp);//這個放裏頭!
}
}
int main()
{
n=read(),K=read();
int sum=0;
for(int i=1; i<=n; ++i) sum+=(A[i]=read());
Aver=1.0*sum/K, Tmp=K*Aver*Aver-2.0*sum*Aver;
// std::random_shuffle(A+1,A+1+n);//這東西。。有時好有時壞
for(int i=1; i<=30; ++i) SA();
printf("%.2lf",sqrt(Ans/(double)K));
return 0;
}
簡單好寫(錯誤率高)的裸隨機化貪心:
使每組(sum_i-Average)盡量平均(也就是使Σ(sum_i^2)最小)。數據範圍這麽小,而且只保留兩位。。
不連續分組很難辦,但是random_shuffle()一下連續分組很多次就可以達到偽不連續分組的效果了。。
具體,我們可以隨便分啊按照某種策略來分,比如依次分給當前sum最小的組。
被隨機數種子一直卡一個點是怎樣的體驗。。我特麽不設了。
//832kb 2448ms
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
const int N=21;
int n,K,A[N],sum[7];
double Ans=1e15,Aver,Tmp;
inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
void Work()
{
std::random_shuffle(A+1,A+1+n);
memset(sum,0,sizeof sum);
for(int i=1; i<=n; ++i)
sum[std::min_element(sum+1,sum+1+K)-sum]+=A[i];
double res=0;
for(int i=1; i<=K; ++i) res+=1.0*sum[i]*sum[i];//Aver可以提出來。。
Ans=std::min(Ans,res+Tmp);
}
int main()
{
n=read(),K=read();
int sum=0;
for(int i=1; i<=n; ++i) sum+=(A[i]=read());
Aver=1.0*sum/K, Tmp=K*Aver*Aver-2.0*sum*Aver;
for(int i=1; i<=300000; ++i) Work();
printf("%.2lf",sqrt(Ans/(double)K));
return 0;
}
BZOJ.2428.[HAOI2006]均分數據(隨機化貪心/模擬退火)