[AT2581] Meaningful Mean [权值树状数组&前缀和]

[$\mathfrak{L}\mathfrak{i}\mathfrak{n}\mathfrak{k}$]

---

首先我们全部减掉 $k$ 那么只需要算区间和非 $0$ 的区间个数
容易想到前缀和，发现可以转化成求前缀和顺序对个数
于是只要用权值数据结构做一下就可以了。

最好不要在 $[0]$ 存东西吧。。我建议还是统一从 $[1]$ 开始的好

---

#include<cstdio>
#include<ctime>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN = 200005;
int n, m;
long long k, ans;
long long a[MAXN];
long long t[MAXN];
long long b[MAXN];
long long Query(int x)
{
	long long ret = 0;
	while(x)
	{
		ret += b[x];
		x -= (x & -x);
	}
	return ret;
}
void Insert(int x)
{
	while (x <= m + 1)
	{
		++b[x];
		x += (x & -x);
	}
}
int main()
{
	scanf("%d%lld", &n, &k);
	for (int i = 1; i <= n; ++i)
	{
		scanf("%lld", &a[i]);
		t[i] = (a[i] += a[i-1] - k);
	}
	sort(t, t + n + 1);
	m = unique(t, t + n + 1) - t + 1;
	for (int i = 0; i <= n; ++i)
	{
		a[i] = lower_bound(t, t + m, a[i]) - t + 1;
		ans += Query(a[i]);
		Insert(a[i]);
	}
	printf("%lld", ans);
	return 0;
}