k小数查询(前缀和)

Posted 2021-09-09 Harris-H

k小数查询(前缀和)

把小于等于$x$的数置为$1$，则答案求区间内包含$x$且$1$的个数为$k$的区间。

先可以找到$x$的位置$pos$，同时预处理$1$的前缀和。

然后开一个桶，用来储存$[pos,n]$范围前缀和的个数。

然后枚举区间$l\in [1,pos]$。

每次的贡献就是：$b[pre[i-1]+k]$。

时间复杂度：$O(n)$

// Problem: k小数查询
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/11177/B
// Memory Limit: 524288 MB
// Time Limit: 2000 ms
// Date: 2021-08-22 20:27:31
// --------by Herio--------

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=2e5+5,M=2e4+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define fi first
#define se second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define ios ios::sync_with_stdio(false),cin.tie(0) 
void Print(int *a,int n){
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\\n",a[n]); 
}
int a[N],b[N];
int main(){
	int n,x,k,pos;scanf("%d%d%d",&n,&x,&k);
	ll ans=0;
	for(int i=1;i<=n;i++){
		int y;scanf("%d",&y);
		if(y==x) pos=i;
		a[i]=a[i-1]+(y<=x);
	}
	for(int i=pos;i<=n;i++) b[a[i]]++;
	for(int i=1;i<=pos;i++) ans+=b[a[i-1]+k];
	printf("%lld\\n",ans);
	return 0;
}