BZOJ4408: [Fjoi 2016]神秘数

考虑处理单组询问
一开始ans=1，每次统计<=ans的数的和sum，若sum>=ans，就用sum+1更新ans
最坏情况下，ans的更新是1，2，3，5，8….是斐波拉契数列
所以ans最多更新log次
用主席树资瓷统计区间<=ans数的和
每次询问log^2
总复杂度mlog^2

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

inline void read(int &x)
{
    char c; while(!((c=getchar())>='0'&&c<='9'));
    x=c-'0';
    while((c=getchar())>='0'&&c<='9') (x*=10)+=c-'0';
}
const int maxn = 110000;

int n,m,u;
int a[maxn];

struct segment{int lc,rc,c;}seg[maxn*40]; int cnt,root[maxn];
int loc;
void upd(int &x,const int l,const int r)
{
    if(!x) x=++cnt;
    seg[x].c+=loc;
    if(l==r) return;
    int mid=l+r>>1;
    if(loc<=mid) upd(seg[x].lc,l,mid);
    else upd(seg[x].rc,mid+1,r);
}
void merge(int &x,int y)
{
    if(!y) return;
    if(!x) { x=y; return; }
    seg[x].c+=seg[y].c;
    merge(seg[x].lc,seg[y].lc);
    merge(seg[x].rc,seg[y].rc);
}
int ans,temp;
void query(int x,int y,const int l,const int r)
{
    if(!y) return;
    if(r<=loc) { temp+=seg[y].c-seg[x].c; return; }
    if(loc<l) return;
    int mid=l+r>>1;
    query(seg[x].lc,seg[y].lc,l,mid);
    query(seg[x].rc,seg[y].rc,mid+1,r);
}

int main()
{
    read(n); 
    for(int i=1;i<=n;i++) 
    {
        read(a[i]);
        if(u<a[i]) u=a[i];
    } u++;
    for(int i=1;i<=n;i++)
    {
        loc=a[i];
        upd(root[i],1,u);
        merge(root[i],root[i-1]);
    }
    read(m);
    while(m--)
    {
        int l,r; read(l); read(r); --l;
        ans=0,temp=0;
        while(ans<=temp)
        {
            ans=temp+1; temp=0; loc=ans;
            query(root[l],root[r],1,u);
        }
        printf("%d\n",ans);
    }

    return 0;
}