bzoj4408 [Fjoi 2016]神秘数（主席树维护前缀和）

如果神秘数现在是ans，统计一下[1,ans]所有在这段区间中出现过的数的sum，如果sum< ans 那么ans肯定永远表示不出了，他就是答案；如果sum>=ans,那么ans是可以得到的，那么把神秘数改为sum+1，继续尝试。建权值线段树，用主席树维护区间前缀和信息即可

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 100010
using namespace std;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,a[N],mx=0,owo=0,rt[N];
struct node{
    int sum,lc,rc;
}tree[40*N];
inline void change(int &p,int l,int r,int k){
    tree[++owo]=tree[p];p=owo;tree[p].sum+=k;
    if(l==r) return;int mid=l+r>>1;
    if(k<=mid) change(tree[p].lc,l,mid,k);
    else change(tree[p].rc,mid+1,r,k);
}
inline int query(int p1,int p2,int l,int r,int x,int y){
    if(x<=l&&r<=y) return tree[p2].sum-tree[p1].sum;
    int mid=l+r>>1,res=0;
    if(x<=mid) res+=query(tree[p1].lc,tree[p2].lc,l,mid,x,y);
    if(y>mid) res+=query(tree[p1].rc,tree[p2].rc,mid+1,r,x,y);
    return res;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();for(int i=1;i<=n;++i) a[i]=read(),mx=max(mx,a[i]);
    for(int i=1;i<=n;++i){
        rt[i]=rt[i-1];change(rt[i],1,mx,a[i]);
    }m=read();
    while(m--){
        int x=read(),y=read();int ans=1;
        while(1){
            int sum=query(rt[x-1],rt[y],1,mx,1,ans);
            if(sum<ans) break;ans=sum+1;
        }printf("%d\n",ans);
    }return 0;
}