牛客练习赛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 i )
休息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<bits/stdc++.h>
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;
 
}