loj2174 「FJOI2016」神秘数

先考虑一下一个集合怎么用 \(O(n)\) 时间求出来，然后用主席树推广到一个序列就可以了。大致思想就是考虑一个数的权值和它前面的数的和的关系。

#include <algorithm>
#include <iostream>
#include <cstdio>
using namespace std;
int n, a[100005], cnt, b[100005], m, uu, vv, rot[100005], tot;
struct Node{
    int l, r, sum;
}nd[5000005];
int build(int l, int r){
    int re=++tot;
    if(l==r)    ;
    else{
        int mid=(l+r)>>1;
        if(l<=mid)  nd[re].l = build(l, mid);
        if(mid<r)   nd[re].r = build(mid+1, r);
    }
    return re;
}
int update(int pre, int l, int r, int x){
    int re=++tot;
    nd[re] = nd[pre];
    nd[re].sum += b[x];
    if(l==r)    ;
    else{
        int mid=(l+r)>>1;
        if(x<=mid)  nd[re].l = update(nd[pre].l, l, mid, x);
        if(mid<x)   nd[re].r = update(nd[pre].r, mid+1, r, x);
    }
    return re;
}
int query(int pre, int now, int l, int r, int x, int y){
    if(b[l]>=x && b[r]<=y)
        return nd[now].sum-nd[pre].sum;
    else if(l==r)
        return 0;
    else{
        int mid=(l+r)>>1;
        int re=0;
        if(x<=b[mid])   re += query(nd[pre].l, nd[now].l, l, mid, x, y);
        if(b[mid]<y)    re += query(nd[pre].r, nd[now].r, mid+1, r, x, y);
        return re;
    }
}
int getAns(int uu, int vv){
    int sum=0, lst=0;
    while(true){
        int tmp=query(rot[uu-1], rot[vv], 1, cnt, lst+1, sum+1);
        if(!tmp)
            return sum+1;
        lst = sum + 1;
        sum += tmp;
    }
}
int main(){
    cin>>n;
    for(int i=1; i<=n; i++){
        scanf("%d", &a[i]);
        b[i] = a[i];
    }
    sort(b+1, b+1+n);
    cnt = unique(b+1, b+1+n) - b - 1;
    for(int i=1; i<=n; i++)
        a[i] = lower_bound(b+1, b+1+cnt, a[i]) - b;
    rot[0] = build(1, cnt);
    for(int i=1; i<=n; i++)
        rot[i] = update(rot[i-1], 1, cnt, a[i]);
    cin>>m;
    while(m--){
        scanf("%d %d", &uu, &vv);
        printf("%d\n", getAns(uu, vv));
    }
    return 0;
}