Interval

题目描述

才发现编故事背景那么难，原谅当初语文考满分的我。对于区间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;
}