luogu P2717 寒假作业

题面传送门
主要一个方法以前没有接触过现在补一下。
就是判断一个区间平均数有没有到\(k\)则每个减\(k\)看总和有没有到\(0\)。
处理完之后就是求正序对数量分治随便搞就好了。
代码实现:

#include<cstdio>
using namespace std;
int n,m,k,x,y,z,a[100039],b[100039];
long long ans;
inline void solve(int x,int y){
	if(x==y) return;
	int m=x+y>>1,l=x,i,head=x-1;
	solve(x,m);solve(m+1,y);
	for(i=m+1;i<=y;i++){
		while(l<=m&&a[l]<=a[i]) b[++head]=a[l++];
		b[++head]=a[i];ans+=l-x;//printf("%d %d\n",a[i],ans);
	}
	while(l<=m) b[++head]=a[l++];
	for(i=x;i<=y;i++) a[i]=b[i];
}
int main(){
	freopen("1.in","r",stdin);
	register int i;
	scanf("%d%d",&n,&k);
	for(i=1;i<=n;i++) scanf("%d",&a[i]),a[i]+=a[i-1]-k;
	solve(0,n);
	printf("%lld\n",ans);
}