题目描述
给定一个长度为 n 的整数数列 和两个整数 x,y 。
请你判断一共有多少个数对 (l,r) 同时满足以下条件:
- 1≤l≤r≤n
- al+al+1+…+ar−1+ar≤x+y×(r−l+1)
输入描述:
第一行输入三个整数 n,x,y
第二行输入 n 个整数 a1,a2,⋯ ,an。
数据范围:
对于60%的数据,1≤n≤2000,−109≤ai≤109,−109≤x,y≤109
对于100%的数据,1≤n≤200000,−109≤ai≤109,−1012≤x,y≤1012
输出描述:
输出一个整数,表示满足条件的数对的数量。
示例1
输入
5 3 3 4 6 5 3 7
输出
5
说明
满足要求的数对一共5组,分别是(1,1)、(2,2)、(3,3)、(4,4)、(3,4)。
思路:al+al+1+…+ar−1+ar≤x+y×(r−l+1),即al+al+1+…+ar−1+ar-y×(r−l+1)≤x,相当于是a[i]-y的前缀和的第r项减去第l-1项,那么这就好办了,把所有a[i]都减y,相加构造前缀数组,那么这题就相当于是小于等于a[i-1]+x且下标大于等于i的数的总和
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
ll a[N];
int main()
{
ll n, x, y;
scanf("%lld%lld%lld", &n, &x, &y);
for (int i = 1; i <= n; i++)
{
scanf("%lld", &a[i]);
a[i] -= y;
}
for (int i = 1; i <= n; i++)
{
a[i] += a[i - 1];
// cout << a[i] << ' ';
}
// cout << endl;
vector<ll> v;
ll ans = 0;
for (int i = n; i >= 1; i--)
{
v.insert(lower_bound(v.begin(), v.end(), a[i]), a[i]);
ans += upper_bound(v.begin(), v.end(), a[i - 1] + x) - v.begin();
// cout << upper_bound(v.begin(), v.end(), a[i] + x) - v.begin() << ' ';
// for (auto t : v)
// {
// cout << t << ' ';
// }
// cout << endl;
}
printf("%lld\n", ans);
return 0;
}