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;
}