平均分

题目：

题目描述
Mech打算申请某个奖学金，需要填写他的成绩。他一共考了n次试，每一次的成绩为ai。他不打算把所有的成绩都填上去，但是为了显得不太假，他至少要保证填上去的成绩是连续的一段aL到aR。为了得到奖学金，他希望交上去的平均成绩大于等于整数P。他想知道，一共有多少种提交成绩的方案可以满足他的要求。

输入输出格式
输入格式：
第一行 n

第二行n个数，ai

第三行 P

输出格式：
一个数，方案数。

输入输出样例
输入样例#1： 
3
1 2 3
3
输出样例#1： 
1
输入样例#2： 
3
1 3 2
2
输出样例#2： 
5
输入样例#3： 
3
1 3 2
3
输出样例#3： 
1
说明
样例解释： 第一个样例：只有{3}可行。

第二个样例：{1,3} {1,3,2} {3} {3,2} {2} 都可行。

对于30%的数据，n≤10000。

对于100%的数据，n≤106，1≤ai≤10^9，1≤P≤10^9。

思路：
考试时挂了的一道好题啊！
由于是求平均分大于p，相当于先把a[i]都减去p，然后求和大于0的子串个数。
此时对a[i]-p处理下前缀和sum[i]，就是相当于求sum[x]-sum[i]>=0的个数(x>i)。
然后离散化下sum，用树状数组处理，相当于求比sum[x]小的数的个数。
此时只用求树状数组中sum[x]前的数的和就好了。

注意：
常数！
在遇到1e6范围的数据时慎用map，set这种STL，会被卡常的。

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 1000000
#define read(x) scanf("%lld",&x)
#define ll long long
#define lowbit(x) (x&-x)

struct Pair{
	ll x;
	int y;
	Pair(){}
	Pair(ll xx,int yy) {x=xx,y=yy;}
	bool operator < (const Pair& oth) const {
		return x<oth.x;
	}
};

int n,m;
int a[maxn+5];
ll sum[maxn+5];

vector<Pair> mp;

int b[maxn+5];

void add(int x) {
	while(x<=n+1) {
		b[x]++;
		x+=lowbit(x);
	}
}

int find(int x) {
	int s=0;
	while(x>0) {
		s+=b[x];
		x-=lowbit(x);
	}
	return s;
}

int main() {
	read(n);
	for(int i=1;i<=n;i++) {
		read(a[i]);
	}
	read(m);
	for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i]-m;
	
	for(int i=0;i<=n;i++) mp.push_back(Pair(sum[i],i));
	sort(mp.begin(),mp.end());
	int c=0;
	for(int i=0;i<mp.size();i++) {
		if(i!=0&&mp[i].x==mp[i-1].x) sum[mp[i].y]=c;
		else sum[mp[i].y]=++c;
	}
	
	ll ans=0;
	for(int i=0;i<=n;i++) {
		ans+=find(sum[i]);
		add(sum[i]);
	}
	
	printf("%lld",ans);
	
	return 0;
}

被卡常代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 1000000
#define read(x) scanf("%lld",&x)
#define ll long long
#define lowbit(x) (x&-x)

int n,m;
int a[maxn+5];
ll sum[maxn+5];

map<ll,int> mp;

int b[maxn+5];

void add(int x) {
	while(x<=n+1) {
		b[x]++;
		x+=lowbit(x);
	}
}

int find(int x) {
	int s=0;
	while(x>0) {
		s+=b[x];
		x-=lowbit(x);
	}
	return s;
}

int main() {
	read(n);
	for(int i=1;i<=n;i++) {
		read(a[i]);
	}
	read(m);
	mp[0]=0;
	for(int i=1;i<=n;i++) {
		a[i]-=m;
		sum[i]=sum[i-1]+a[i];
		mp[sum[i]]=i;
	}
	int c=0;
	for(map<ll,int>::iterator it=mp.begin();it!=mp.end();++it) {
		c++;
		it->second=c;
	}
	for(int i=0;i<=n;i++) sum[i]=mp[sum[i]];
	ll ans=0;
	for(int i=0;i<=n;i++) {
		ans+=find(sum[i]);
		add(sum[i]);
	}
	
	printf("%lld",ans);
	
	return 0;
}