题目:
题目描述
农历新年马上就要到了,奶牛们计划举办一次聚会庆祝新年的到来。但是,奶牛们并不喜欢走太远的路,这会给他们的聚会带来消极情绪,当一头奶牛的消极指数为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表示消极情绪之和的最小值,结果四舍五入为一个整数。
样例输入 Copy
1
5
0.9 2
1.4 4
3.1 1
6.2 1
8.3 2
样例输出 Copy
Case #1: 300
解析:
如果把聚会的地点作为自变量x,那么奶牛的消极情绪和就是关于x的三次方的函数,又由于奶牛和聚会地点的距离不可能是负的,所以要加上绝对值。
x的三次方再加上绝对值,函数图像很明显是如下的凹函数,最低点就是我们要求的那个点
如下图:
和二分一样,先初始化l和r,然后再找到l和r之间的两个三等分点,分别记作lmid和rmid,比较lmid和rmid对应的y值的大小,如果lmid的y值大(如上图所示),说明lmid距离我们要找的最低点比较远,这时候更新l的值(l=lmid),反之,更新r的值。
对于分治法的边界问题
如果是l和r都是整形的情况,那么while(l+1!=r)是万用模板,用其他的写法可能在一些特殊情况下会出现小错误。
如果是像本题一样l和r是浮点型的情况下,只能让l和r尽可能接近:while(right - left >= 1e-8),具体right-left要大于多少,还需具体情况具体分析,不宜太小也不宜太大,就像本题我一开始设置的是1e-16,因为double能存储小数点后16位却编译错误。
为什么本题不建议使用二分
本题还有一种解法就是对原函数求导,对导函数进行二分,找到导函数为0的点就是我们要找的点。
不建议使用二分有以下两点考虑:
1.原函数中具有绝对值,求导函数需要分情况讨论。
2.题目要求输出的是奶牛的负面情绪,即使找到了导数为0的点,也需要再通过原函数计算情绪值,倒不如直接在原函数上进行三分。
代码:
#include<iostream>
#include<cmath>
using namespace std;
const int N = 5e4+5;
double s[N];
double w[N];
int n;
double sum(double x)//求(x-s)三次方*w的累加和
{
double ans = 0;
for(int i=1;i<=n;i++)
{
ans+=pow(abs(x-s[i]),3)*w[i];
}
return ans;
}
void solve(int num)
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>s[i]>>w[i];
double left = -1e6,right = 1e6;
while(right - left >= 1e-8)
{
double lmid = (left+right)/2;
double rmid = (lmid+right)/2;
if(sum(lmid)<sum(rmid))//如果右三分线的y大于左三分线的y,说明左边的三分线比右边的三分线更接近凹函数的凹点,这时候需要更新right值
right = rmid;
else
left = lmid;
}
long long ans = (sum(left)+0.5);//一定要开longlong,#define int long long 会和cmath函数起冲突
cout<<"Case #"<<num<<": "<<ans<<'\n';
}
int main()
{
int ca;
cin>>ca;
for(int i=1;i<=ca;i++)
solve(i);
}