题意:多组输入,长度为n的序列,每个人有属性值为 a i a_i ai,多次询问,询问区间 [ l , r ] [l,r] [l,r],最大的h,使得有起码h个人的属性值也是大于等于h的。
题目链接:C
主席树上判断往左还是往右走就好,优先向右走,否则用一个变量加上右子树,判断左子树。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+7;
struct Tree{
int lc,rc,sum;
}tree[maxn*50];
int root[maxn];
int tot;
int build(int l,int r){
int k=++tot;
tree[k].sum=0;
if(l==r) return k;
int mid=(l+r)>>1;
tree[k].lc=build(l,mid);
tree[k].rc=build(mid+1,r);
return k;
}
int updata(int p,int l,int r,int id){
int k=++tot;
tree[k]=tree[p];
++tree[k].sum;
if(l==r) return k;
int mid=(l+r)>>1;
if(id<=mid) tree[k].lc=updata(tree[p].lc,l,mid,id);
else tree[k].rc=updata(tree[p].rc,mid+1,r,id);
return k;
}
int sum;
int myfind(int p,int q,int l,int r){
if(l==r) return l;
int mid=(l+r)>>1;
if(sum+tree[tree[p].rc].sum-tree[tree[q].rc].sum>=mid+1)
return myfind(tree[p].rc,tree[q].rc,mid+1,r);
sum+=tree[tree[p].rc].sum-tree[tree[q].rc].sum;
return myfind(tree[p].lc,tree[q].lc,l,mid);
}
int main(){
int n,q,x,y;
while(scanf("%d%d",&n,&q)!=EOF){
root[0]=build(1,n);
for(int i=1;i<=n;++i){
scanf("%d",&x);
root[i]=updata(root[i-1],1,n,x);
}
while(q--){
scanf("%d%d",&x,&y);
sum=0;
printf("%d\n",myfind(root[y],root[x-1],1,n));
}
}
return 0;
}