题意:
给出一个长度为n的序列,Q次询问l,r,H,询问在[l,r]区间中小于H的数有多少个?
题解:
用划分树求第k小的,然后二分答案即可
AC代码:
#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
typedef long long LL;
typedef pair<int,int> pii;
const int MAXN = 1e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int num[20][MAXN],val[20][MAXN],a[MAXN],sorted[MAXN];
inline void build(int l,int r,int dep){
if(l==r) return;
int mid=(l+r)>>1,same=mid-l+1;
for(int i=l;i<=r;i++) if(val[dep][i]<sorted[mid]) same--;
int ln=l,rn=mid+1;
for(int i=l;i<=r;i++){
if(val[dep][i]<sorted[mid] || (val[dep][i]==sorted[mid] && same>0)){
val[dep+1][ln++]=val[dep][i];
if(val[dep][i]==sorted[mid]) same--;
}else val[dep+1][rn++]=val[dep][i];
num[dep][i]=num[dep][l-1]+ln-l;
}
build(l,mid,dep+1); build(mid+1,r,dep+1);
}
inline int query(int dep,int l,int r,int ql,int qr,int k){
if(ql==qr) return val[dep][ql];
int mid=(l+r)>>1,tot=num[dep][qr]-num[dep][ql-1];
if(tot>=k){
int ll=l+num[dep][ql-1]-num[dep][l-1];
int lr=ll+tot-1;
return query(dep+1,l,mid,ll,lr,k);
}else{
int lr=qr+num[dep][r]-num[dep][qr];
int ll=lr-(qr-ql+1-tot)+1;
return query(dep+1,mid+1,r,ll,lr,k-tot);
}
}
signed main(){
#ifndef ONLINE_JUDGE
freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
int T,cas=0; scanf("%d",&T);
while(T--){
int n,Q; scanf("%d%d",&n,&Q);
for(int i=1;i<=n;i++) scanf("%d",&val[0][i]),sorted[i]=val[0][i];
sort(sorted+1,sorted+n+1);
build(1,n,0);
printf("Case %d:\n",++cas);
while(Q--){
int l,r,k; scanf("%d%d%d",&l,&r,&k); l++;r++;
int ll=1,rr=r-l+1,ans=0;
while(ll<=rr){
int mid=(ll+rr)>>1;
if(query(0,1,n,l,r,mid)<=k) ll=mid+1,ans=mid;
else rr=mid-1;
}
printf("%d\n",ans);
}
}
return 0;
}