CodeForces1024 페티 및 배열 (CDQ 파티션 / 펜윅 나무)

CodeForces1024 페티 및 배열 (CDQ 파티션 / 펜윅 나무)

문

질문의 의미 :

당신에게 길이 n의 순서를 부여, 당신은 얼마나 많은 서브 인터벌 요청해야 미만 또는 동일 $티$

해결 방법 :

이 질문은 실제로 역에 펜윅 트리를 촉진하기 위해 노력하고있다. 펜윅 트리, 나는 (많은 노크하기 어려운 느낌)을 분할 및 정복 방법은 않습니다 CDQ 사용 할 확실히 가능하다.

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
long long ans=0;
long long t;
long long sum[maxn],temp[maxn];
void cdq(int l,int r)
{
	if(l==r)
	{
		if(sum[l]<t) ans++;
		return;
	}
	int mid=(l+r)>>1;
	cdq(l,mid);cdq(mid+1,r);
	int i=l,j=mid+1,tot=l;
	while(i<=mid||j<=r)
	{
		if(sum[i]<sum[j])
		{
			int num=lower_bound(sum+j+1,sum+r+1,sum[i]+t)-sum-j-1;
			ans+=num+1;
			if(sum[j+num]>=sum[i]+t) ans--;
			temp[tot++]=sum[i++];
		}
		else
		{
			int num=upper_bound(sum+i+1,sum+mid+1,sum[j]-t)-sum-i-1;
			ans+=(mid-i+1-num);
			if(sum[num+i]+t<=sum[j]) ans--;
			temp[tot++]=sum[j++];
		}
		if(i>mid)
		{
			while(j<=r)
			{
				temp[tot++]=sum[j++];
			}
		}
		else if(j>r)
		{
			while(i<=mid)
			{
				temp[tot++]=sum[i++];
			}
		}
	}
	for(i=l;i<=r;i++)
		sum[i]=temp[i];
}
int main()
{
	#ifdef TEST
		freopen("input.txt","r",stdin);
    #endif
	int T,n,m,i,j,k;
	scanf("%d%lld",&n,&t);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&k);
		sum[i]=sum[i-1]+k;
	}
	cdq(1,n);
	cout<<ans<<endl;
}