模拟退火

来源:互联网 发布:linux db2创建schema 编辑:程序博客网 时间:2024/04/29 04:39

题目链接
题意:给出椭球的6个参数,求椭球上一点,到原点的距离最短。
别人讲的模拟退火

#include<cstdio>#include<cstring>#include<cmath>#include<vector>#include<iostream>#include<algorithm>using namespace std;const double inf=0x3f3f3f3f;const int maxn=1e4+10;const double eps=1e-8;const double cooling=0.99;//降温速度double a,b,c,d,e,f;int dx[10]={0,0,1,1,1,-1,-1,-1};int dy[10]={1,-1,0,-1,1,0,-1,1};double get_dis(double x,double y,double z)//计算到原点的距离{    return sqrt(x*x+y*y+z*z);}double get_z(double x,double y)//根据x和y计算z的值{    double C=a*x*x+b*y*y+f*x*y-1,B=e*x+d*y,A=c;    double s=B*B-4*A*C;    if(s<0) return inf;    double z1=(0-B+sqrt(s))/(2.0*A);    double z2=(0-B-sqrt(s))/(2.0*A);    return fabs(z1)>fabs(z2)?z2:z1;}double solve()//模拟退火{    double step=1;//步长    double x=0,y=0,z;    while(step>eps)    {        z=get_z(x,y);        for(int i=0;i<8;i++)//8个方向走        {            double nx=x+dx[i]*step;            double ny=y+dy[i]*step;            double nz=get_z(nx,ny);            if(get_dis(x,y,z)>get_dis(nx,ny,nz))//找到更优解就更新                x=nx,y=ny,z=nz;        }        step*=cooling;//退火    }    return get_dis(x,y,z);//返回最优解}int main(){    while(~scanf("%lf%lf%lf%lf%lf%lf",&a,&b,&c,&d,&e,&f))    {        printf("%.7lf\n",solve());    }    return 0;}
0 0