P2717 寒假作业

$Hyperlink$

https://www.luogu.org/problem/P2717

---

$Description$

给定一个长度为$n$的序列$A$，求区间平均值不小于$k$的连续子序列个数

$n\le 10^5,A_i\le 10^4,k\le 10^4$

---

$Solution$

设选择的区间是$i\sim j$，这里$i<j$，则

$$\frac{{\sum }_{p=i}^j{a}_p}{j-i+1}\ge k$$

移项

$$\sum _{p=i}^j{a}_p\ge k(j-i+1)$$

即

$$\sum _{p=i}^j{a}_p\ge \sum _{p=i}^{j}k$$

移项

$$\sum _{p=i}^j(a_p-k)\ge 0$$

令$a_p=a_p-k$，$s_i={\sum }^{j<=i}{a}_j$

于是
$$s[j]-s[i-1]\ge 0$$

于是
$$s[j]\ge s[i-1]$$

于是树状数组求顺序对，复杂度$O(nlogn)$

细节：

1. 由于$a_i-k$可能小于0，而树状数组莫得小于0的操作，所以要离散化，这也是后面那个$log$的上限是$10^5$而不是$10^4$的原因
2. 注意求顺序对的时候是包括$s[0]$的

---

$Code$

#include<cstdio>
#include<cctype>
#include<algorithm>
#define LL long long
#define lb(x) x&-x
using namespace std;int n,k,a[100001],c[100001],s[100001],b[100001];
LL ans; 
inline void add(int x){for(;x<=n;x+=lb(x)) c[x]++;return;}
inline int get(int x){int s=0;for(;x;x-=lb(x)) s+=c[x];return s;}
inline LL read()
{
    char c;LL f=0,d=1;
    while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
    while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
    return d*f;
}
signed main()
{
	n=read();k=read();
	for(register int i=1;i<=n;i++) a[i]=read()-k,b[i]=s[i]=s[i-1]+a[i];
	sort(b,b+1+n);
	int tot=unique(b,b+1+n)-b-1;
	for(register int i=0;i<=n;i++) s[i]=lower_bound(b,b+1+tot,s[i])-b+1;
	add(s[0]);
	for(register int i=1;i<=n;i++){ans+=get(s[i]);add(s[i]);}
	printf("%lld",ans);
}