WOJ 1718 FloatingMedian [树状数组] [区间第K大]

传送门

其实就是动态维护一个树状数组 , 然后中位数就是区间第(k+1)/2 大

我们可以对树状数组二进制拆分 , 如果当前答案加上枚举的这个区间的值不超过 (k+1)/2 

就加上这个答案 , 复杂度是logn 的

#include<bits/stdc++.h>
#define LL long long
#define Mod 65536
#define Next(x) (x*mul%Mod+add)%Mod
using namespace std;
int mul,add,n,k; LL ans;
int c[Mod+1],a,b,seed; 
void Up(int x,int val){for(;x<=Mod;x+=x&-x) c[x] += val;}
int Find(int x){
	int ans = 0 , cnt = 0;
	for(int i=16;i>=0;i--){
		ans += 1<<i;
		if(ans>Mod || cnt + c[ans] >= x) ans -= 1<<i;
		else cnt += c[ans];
	} return ans+1;
}
int main(){
	scanf("%d%d%d%d%d",&seed,&mul,&add,&n,&k); 
	a = b = seed; int mid = (k+1) / 2;
	for(int i=1;i<=n;i++){
		Up(a+1,1); a = Next(a); if(i>=k){
			ans += Find(mid) - 1;
			Up(b+1,-1); b = Next(b);
		}
	} printf("%lld",ans); return 0;
}