牛客练习赛6 A 猴子吃香蕉(等差数列)

题目链接:https://www.nowcoder.com/acm/contest/26/A

题目描述

有n只猴子,第i只猴子每过xi小时会连续吃香蕉yi小时。猴子从第二次开始每次休息结束后这只猴子连续吃香蕉的时间会增加zi小时。

给定n只猴子,每一只的x i,y i,z i,以及时间t,求在前t小时中,所有猴子共吃了多少小时。 对于一只猴子来说是这样的:
从第1小时开始: 
休息x i小时( 1 -> x i ) 
吃y i小时( x i + 1 -> x i + y )
休息x i小时
吃y i+z i小时
休息x i小时
吃y i+z i+z i小时
......

输入描述:

第一行两个数n和t;

之后n行,第i+1行每行三个数xi,yi,zi.

输出描述:

一行一个数表示答案. 示例1

输入

10 100000000 1 0 0 1 0 5 1 2 2 1 2 8 1 3 0 1 5 0 1 5 2 1 5 5 1 7 0 1 8 3

输出

845787522

说明

每只猴子分别吃了0,99993675,99990000,99994999,75000000,83333333,99990002,99993676,87500000,99991837小时 示例2

输入

1 233333 233 233 233

输出

223081

说明

那只猴子吃了223081小时

备注:

对于100%的数据: 0 <= n <= 100000 0 <= t <= 2000000000 x + y + z > 0 0 <= x , y , z <= 2000000000

解析:如果直接暴力的话,必然会超时,所以肯定有O(n)或者O(n*log(n))的方法,其实仔细一想,这个是一个等差数列,公差d = z[i], a1 = x[i] + y[i],an = a1 + (n-1)*d,然后就是求前n项的和Sn = t,这里不知道n,列一个一元二次的方程解除n然后计算时间即可

代码:

#include
using namespace std;typedef long long LL;//d = z
//a1 = x+y
//a2 = a1+d
//
//an = a1 + (n-1)*d
//
//2*t = 2sn = n*(a1+an) = n*(2a1+(n-1)*d)
//ax^2+bx+c=0
//
//2a1*n+d*n^2-dn-2*t = 0
//
//d*n^2 + (2a1-d)n + (-2t) = 0
//
//a = d;
//b = 2a1-d
//c = -2t
//
//x =
//
//
//ans = n*x
//LL x[100009], y[100009], z[100009];int main()
{LL n, t;scanf("%lld%lld", &n, &t);for(int i = 1; i <= n; i++) scanf("%lld%lld%lld", &x[i], &y[i], &z[i]);long long ans = 0, a, b, c, X, cur;for(int i = 1; i <= n; i++){cur = t;if(!z[i]){LL now = 0;now += cur/(x[i]+y[i])*y[i];cur %= (x[i]+y[i]);cur -= x[i];if(cur > 0) now += cur;ans += now;continue;}a = z[i];b = 2ll*(x[i]+y[i])-a;c = -2ll*t;X = (long long)((-b+sqrt(b*b-4.0*a*c))/2.0/a + 1e-8);if(X > 0){LL now = 0;LL k = X*(x[i]+y[i]+x[i]+y[i]+z[i]*(X-1))/2;now += k - X*x[i];cur -= k;cur -= x[i];if(cur > 0) now += cur;ans += now;}}printf("%lld\n", ans);return 0;}



本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部