题目描述
才发现编故事背景那么难,原谅当初语文考满分的我。对于区间L,R而言,它的长度是R−L+1,区间和则是a[L]+…+a[R]。(如果用数组a去记录)
注意:区间元素的下标是从1开始的。
如:序列(5,2,1,1,3,1,4)
区间[2,4]的长度是3,区间[2,5]的区间和为2+1+1+3=7。
现在Ocean用程序随机生成了一个序列,该序列一共有N个整数。 对于这个序列,他想知道区间长度>=L且满足区间和>=M的不同区间有多少个?
对于两个区间[x1,y1]和[x2,y2],若x1!=x2或者 y1!=y2 则可以认为两个区间是不同的。
输入
第一行输入一个整数T ,代表有T组测试数据。 每组数据占两行,第一行依次输入三个整数N,L,M分别代表上面提到的信息。
下面一行输入N个整数ai。
注:1<=T<=100,1<=L<=N<=10000,−100000<=M,ai<=100000。
输出
对每组测试数据,输出一个整数代表最后的结果。
样例输入
3
1 1 1
1
4 2 3
1 2 3 4
6 3 -7
-1 -2 -3 4 -5 6
样例输出
1
6
10
AC码:
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
vector <long long> v;
long long sum[10010];
int main()
{
int T,a,N,M,L;
scanf("%d",&T);
while(T--)
{
v.clear() ;
scanf("%d %d %d",&N,&L,&M);
for(int i=1;i<=N;i++)
{
scanf("%d",&a);
sum[i]=sum[i-1]+a;
}
long long ans=0,l=1,r=L;
v.push_back(0);
while(r<=N)
{
long long s=sum[r]-M;
int fr=upper_bound(v.begin(),v.end() ,s)-v.begin() ;//返回键值>给定元素的第一个位置
ans+=fr;
int fl=lower_bound(v.begin(),v.end() ,sum[l])-v.begin() ;
v.insert(fl+v.begin() ,sum[l]);
++l;
++r;
}
printf("%lld\n",ans);
}
return 0;
}