HDU 4355 Party All the Time (三分求凸函数极值)

最新推荐文章于 2021-09-30 14:53:46 发布

weixin_30565327

最新推荐文章于 2021-09-30 14:53:46 发布

阅读量54

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/AbandonZHANG/archive/2013/04/04/4114232.html

本文介绍了一种使用三分法求解一维精灵聚会问题的算法，目标是最小化所有精灵到选择点的能量消耗。通过分析函数性质，实现高效的求解过程，并提供了HDU4355代码实例。

题目大意：有n个精灵位于一维坐标轴上，每个精灵有一个权值w，每个精灵走到另一个位置耗费能量S^3 * w。（s是两点间距离）。现在精灵要聚会，求选取一点，使所有精灵到这点浪费能量之和最小。分析：设这点的位置是x，精灵位置是pi，则浪费能量值和f = sigma[(pi-x)^3*wi].其中x是变量. 通过对这个函数求二次导可以发现，这个函数是一个下凸函数.可以用三分做. 具体的三分方法如图：凸函数：

凹函数：

三分求凸函数极值模板：

//如果一个解函数的二次导恒>0(or <0)，则该函数为凸函数.

//精度模板
const double eps = 1e-6;
bool dy(double x,double y)  {   return x > y + eps;} 		// x > y
bool xy(double x,double y)  {   return x < y - eps;} 		// x < y
bool dyd(double x,double y) {   return x > y - eps;} 		// x >= y
bool xyd(double x,double y) {   return x < y + eps;}     	// x <= y
bool dd(double x,double y)  {   return fabs( x - y ) < eps;}    // x == y

double cal(double pos){
    /* 根据题目的意思计算 */
}

double solve(double left, double right){
    while(xy(left, right)){
        double mid = DMID(left, right);
        double midmid = DMID(mid, right);
        double mid_area = cal(mid);
        double midmid_area = cal(midmid);
        if (xyd(mid_area, midmid_area))     //下凸函数，上凸函数用dyd(mid_area, midmid_area)
            right = midmid;
        else
            left = mid;
    }
    return left;
}

HDU 4355 代码：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define MID(x,y) ((x+y)>>1)
#define DMID(x,y) ((x+y)/2)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;

const int N = 50002;
double x[N], w[N];
int n;

//二分精度模板
const double eps = 1e-6;
bool dy(double x,double y)  {   return x > y + eps;} 		// x > y
bool xy(double x,double y)  {   return x < y - eps;} 		// x < y
bool dyd(double x,double y) {   return x > y - eps;} 		// x >= y
bool xyd(double x,double y) {   return x < y + eps;}     	// x <= y
bool dd(double x,double y)  {   return fabs( x - y ) < eps;}    // x == y

double cal(double pos){
    /* 根据题目的意思计算 */
    double res = 0.0;
    for (int i = 0; i < n; i ++){
        res += abs(pos - x[i]) * abs(pos - x[i]) * abs(pos - x[i]) * w[i];
    }
    return res;
}

double solve(double left, double right){
    while(xy(left, right)){
        double mid = DMID(left, right);
        double midmid = DMID(mid, right);
        double mid_area = cal(mid);
        double midmid_area = cal(midmid);
        if (xyd(mid_area, midmid_area))     //下凸函数，上凸函数dyd(mid_area, midmid_area)
            right = midmid;
        else
            left = mid;
    }
    return left;
}

int main(){
    int t;
    scanf("%d", &t);
    for (int caseo = 1; caseo <= t; caseo ++){
        scanf("%d", &n);
        double low = -1000000, high = 1000000;
        for (int i = 0; i < n; i ++){
            scanf("%lf %lf", &x[i], &w[i]);
            low = min(low, x[i]);
            high = max(high, x[i]);
        }

        printf("Case #%d: %.0f\n", caseo, (cal(solve(low, high))));
    }
	return 0;
}

转载于:https://www.cnblogs.com/AbandonZHANG/archive/2013/04/04/4114232.html