Codeforces Round #510 (Div. 2), problem (D) Petya and Array 分治

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Andy123321aa/article/details/83590255

本题如果采用两层for循环遍历每个l和r，时间复杂度为O(n2)，将会超时。所以可以采用分治的算法，将数组二分递归，完成一个分支之后将此次l和r之间的数进行排序，目的是能以O(n)的时间复杂度统计出此分支符合要求的个数。

ac代码：

#include <bits/stdc++.h>
#define FOR(I,A,B) for(int I = (A); I < (B); I++)
#define FORE(I,A,B) for(int I = (A); I <= (B); I++)
#define PRII pair<int,int> 
#define LL long long 
#define INF 1000000001

using namespace std;
int n;
LL a[200005],t;
LL ans;
void dg(int l, int r){
	if(l==r) return;
	int mid=(l+r)/2;
	dg(l,mid);
	dg(mid+1,r);
	for(int i=l,j=mid;i<=mid;i++){
		while(j<r && (a[j+1]-a[i])<t) j++;
		ans+=j-mid;
	}
	sort(a+l,a+r+1);
}
int main()
{
	cin>>n>>t;
	FORE(i,1,n){
		cin>>a[i];
		a[i]+=a[i-1];	
	} 
	dg(0,n);
	cout<<ans<<endl;
    return 0;
}