1. 程式人生 > >bzoj2428 [HAOI2006]均分資料

bzoj2428 [HAOI2006]均分資料

一開始想了一個用優先佇列的貪心,只能在洛谷過三個點(我還是too young too simple)。。。所以在網上看了一下

模擬退火

模擬退火之前在網上也看到過這個演算法吧,但是一直都沒有學,覺得難就沒學,其實仔細看一下還是比較簡單的

這題就是模擬退火的一個裸題,先上程式碼,詳細的後面再補


模擬退火就是模擬物理上物體從高溫到低溫的一個過程,具體的網上也很多,兔子的那個比喻就很恰當

高溫的時候物體內部的粒子比較混亂,所以它所在的地方可能性很多(大概是這個意思吧。。),那麼在程式裡面高溫的時候狀態轉移過去的概率就比較大

反之,低溫的時候物體內部粒子傾向於穩定,那麼這個時候轉移的概率比較小,除非當前狀態比之前更優,那麼直接模擬了


//Decision's template
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<vector>
#include<queue>
#include<stack>
#include<algorithm>
#include<string>
#include<cmath>
#include<map>
#include<set>
using namespace std;

#define DP_maxn 16
#define maxn 100000
#define INF 10000007
#define MOD 1000000007
#define mst(s,k) memset(s,k,sizeof(s))

typedef long long ll;

struct Edge{
    int from,to,dist;
    Edge(int u,int v,int d):from(u),to(v),dist(d){}
};

/*-------------------------------template End--------------------------------*/

int a[30],n,k,pos[30];
double minans = (double)INF,ans,ave = 0.0,sum[30];

double sqr(double a) {return a*a;}

void  Fire()
{
	mst(sum,0);
	double r= 0.9,T = 10000.00;
	ans = 0.0;
	for(int i = 1;i<=n;i++)
	{
		pos[i] = rand()%k+1;
	}
	for(int i = 1;i<=n;i++) sum[pos[i]] += a[i];
	for(int i = 1;i<=k;i++)
	{
		ans+=(sum[i]-ave)*(sum[i]-ave);
	}
	//cout<<ans<<endl;
	while(T>0.1)
	{
		T*=r;
		int t = rand()%n+1;
		int x = pos[t],y;
		if(T>500) y = min_element(sum+1,sum+k+1)-sum;
		else y = rand()%k+1; 
		if(x==y) continue;
		double tmp = ans;
		ans -= (sum[x]-ave)*(sum[x]-ave);
		ans -= (sum[y]-ave)*(sum[y]-ave);
		sum[x]-=a[t];sum[y]+=a[t];
		ans += (sum[x]-ave)*(sum[x]-ave);
		ans += (sum[y]-ave)*(sum[y]-ave);
		if(ans<=tmp||rand()%10000<=T){
			pos[t] = y;
		} 
		else{
			ans = tmp;
			sum[x]+=a[t];
			sum[y]-=a[t];
		}
	}
	minans = min(ans,minans);
	//cout<<minans<<endl;
}

int main()
{
	//freopen("std.in","r",stdin);
	//freopen("std.out","w",stdout);
	mst(sum,0);
	cin>>n>>k;
	for(int i = 1;i<=n;i++)
	{
		cin>>a[i];
		ave+=a[i];
		swap(a[i],a[rand()%i+1]);
	}
	ave /= k;
	for(int i = 1;i<=10000;i++) Fire();
	printf("%.2f\n",sqrt(minans/k));
	return 0;
}