算法导论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表示消极情绪之和的最小值,结果四舍五入为一个整数。

样例输入 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);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

owooooow

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值