HDU 5017 Ellipsoid 模拟退火第一题

为了补这题，特意学了下模拟退火算法，感觉算法本身不是很难，就是可能降温系数，步长等参数不好设置。

具体学习可以参见： http://www.cnblogs.com/heaad/archive/2010/12/20/1911614.html 我认为讲的很不错，通俗易懂。

这题设置一个step为1，降温系数为0.99，因为系数越大，得到最优解的概率越大，虽然可能会慢一点。因为是三维的，所以往八个方向扩展找邻域解，然后遇到比他优的解一定接受。

代码：（参照网上代码）

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#define Mod 1000000007
#define eps 1e-8
using namespace std;

int dx[8] = {0,0,1,-1,1,-1,1,-1};
int dy[8] = {1,-1,0,0,1,1,-1,-1};
double a,b,c,d,e,f;

double dis(double x,double y,double z)
{
return sqrt(x*x + y*y + z*z);
}

double calc(double x,double y)
{
double A = c;
double B = d*y+e*x;
double C = f*x*y + a*x*x + b*y*y - 1.0;
double delta = B*B-4.0*A*C;
if(delta < 0.0) return Mod+10.0;    //不在椭球上
delta = sqrt(delta);
double soz1 = (-B + delta)/(2.0*A);
double soz2 = (-B - delta)/(2.0*A);
if(dis(x,y,soz1) < dis(x,y,soz2))
return soz1;
return soz2;
}

double Simulated_Annealing()
{
double x = 0,y = 0,z = sqrt(1.0/c);  //当前最优解
double step = 1.0, rate = 0.99;
while(step > eps)
{
for(int k=0;k<8;k++)
{
double kx = x + step*dx[k];
double ky = y + step*dy[k];
double kz = calc(kx,ky);
if(kz >= Mod) continue;
if(dis(kx,ky,kz) < dis(x,y,z))
{
x = kx,y = ky,z = kz;
}
}
step *= rate;
}
return dis(x,y,z);
}

int main()
{
while(scanf("%lf%lf%lf%lf%lf%lf",&a,&b,&c,&d,&e,&f)!=EOF)
{
printf("%.7f\n",Simulated_Annealing());
}
return 0;
}

View Code