中国矿业大学算法概论作业一I、奶牛的聚会

I、奶牛的聚会

题目描述

农历新年马上就要到了，奶牛们计划举办一次聚会庆祝新年的到来。但是，奶牛们并不喜欢走太远的路，这会给他们的聚会带来消极情绪，当一头奶牛的消极指数为Wi，他参加聚会所需行走的距离为si，那么他就会给聚会带来Si3*Wi的消极情绪。所有奶牛所在位置都在一条直线上，已知所有奶牛的坐标和消极指数，求如何确定聚会地点，使得所有奶牛给聚会带来的消极情绪之和最小，输出消极情绪之和的最小值。

输入

第一行包含一个整数 Ca(Ca<=20) ，表示有 Ca 组测试数据。
对于每组测试数据：第一行包含一个整数n(1<=n<=50000) ，表示奶牛的数量。接下来 n 行每行包含两个浮点数Si和wi (-106<=Si<=106, 0<Wi<15)。

输出

对于每组测试数据，输出 “Case #c: ans” ，其中c表示测试数据编号，ans表示消极情绪之和的最小值，结果四舍五入为一个整数。

样例输入

1
5
0.9 2
1.4 4
3.1 1
6.2 1
8.3 2

样例输出

Case #1: 300

题解(dp问题 )

si^3*wi是凹函数；
要求的sum是n个凹函数的和；就是一个凹函数求最小值，我们可以用三分；
或者求导用二分；

#include <iostream>
#include<cmath> 
typedef long long ll;
using namespace std;
double *s = NULL;
double *w = NULL;
int x;
// 定义的函数计算情绪之和 
double F(double t){
	double sum = 0;
	for(int i=0;i<x;i++){
		sum += pow(abs(s[i]-t), 3)*w[i];
	}
	return sum;
}
int main(){
	int n, k;
	cin>>n; // n组测试数据 
	k = n;
	while(k--){
		cin>>x; // x个数据 
		s = new double[x];
		w = new double[x];	
		double l = 1000000, r = -1000000;
		for(int i=0;i<x;i++){
			cin>>s[i]>>w[i];
			r = max(s[i], r); // r 是坐标轴上的最小值 
			l = min(s[i], l); // l 是坐标轴上的最大值 
		}
		while(r-l>0.000001){
			double x1 = l + (r-l)/3; // x1取 [l, r]区间的1/3 点 
			double x2 = r - (r-l)/3; // x2取 [l, r]区间的2/3 点 
			if(F(x1) > F(x2)) l = x1; // 若x1点的情绪之和较大， 则x1替换l 
			else r= x2;
		}  	
		cout<<"Case #"<<n-k<<": "<< ll(F(l) + 0.5)<<endl;
	} 
	delete []s;
	delete []w;
	return 0;
}