220722 T4 求和 /P4587 [FJOI2016]神秘数 （主席树）

好久没打主席树了，都忘了怎么用了......

假设我们选了一些数能构成[0,x]范围内的所有值，下一个要加的数是k（k<=x+1），那么可以取到[0,x+k]内的所有取值，所以有一种做法：

对于每个询问区间，先排序，依次加入k，k<=x+1时，取到[0,x+k]；k>x+1，则x+1不能取到，就是答案。复杂度O（mnlogn）。

更优做法：

[0,x]拼成，令ans=x+1，统计在[1,x+1]范围内的数之和sum，如果sum>=ans，说明肯定有一个未加入的数在[1，x+1]之内，又令ans=sum+1，继续；否则答案就是ans。

ai之和在1e9之内，用主席树统计sum。复杂度O(n logn log(∑ai​))。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10,INF=1e9;
int n,m,a[N],rt[N],tot=0;
struct node{
    int lc,rc,v;
}t[N<<5];

void insert(int &i,int j,int l,int r,int pos,int k){
    t[i=++tot]=t[j],t[i].v+=k;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(pos<=mid) insert(t[i].lc,t[j].lc,l,mid,pos,k);
    else insert(t[i].rc,t[j].rc,mid+1,r,pos,k);
}

int query(int i,int j,int l,int r,int ql,int qr){
    if(ql<=l && qr>=r) return t[j].v-t[i].v;
    int mid=(l+r)>>1,res=0;
    if(ql<=mid) res+=query(t[i].lc,t[j].lc,l,mid,ql,qr);
    if(qr>mid) res+=query(t[i].rc,t[j].rc,mid+1,r,ql,qr);
    return res;
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        insert(rt[i],rt[i-1],1,INF,a[i],a[i]);
    }
    scanf("%d",&m);
    while(m--){
        int l,r,ans=1;scanf("%d%d",&l,&r);
        while(1){
            int res=query(rt[l-1],rt[r],1,INF,1,ans);
            if(res>=ans) ans=res+1;
            else break;
        }
        printf("%d\n",ans);
    }
    return 0;
}

 

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。