bzoj4408: [Fjoi 2016]神秘数【主席树】

Description

一个可重复数字集合S的神秘数定义为最小的不能被S的子集的和表示的正整数。例如S={1,1,1,4,13}，

1 = 1

2 = 1+1

3 = 1+1+1

4 = 4

5 = 4+1

6 = 4+1+1

7 = 4+1+1+1

8无法表示为集合S的子集的和，故集合S的神秘数为8。

现给定n个正整数a[1]..a[n]，m个询问，每次询问给定一个区间l,r，求由a[l],a[l+1],…,a[r]所构成的可重复数字集合的神秘数。

Input

第一行一个整数n，表示数字个数。
第二行n个整数，从1编号。
第三行一个整数m，表示询问个数。
以下m行，每行一对整数l,r，表示一个询问。

Output

对于每个询问，输出一行对应的答案。

Sample Input

5

1 2 4 9 10

5

1 1

1 2

1 3

1 4

1 5

Sample Output

2

4

8

8

8

HINT

对于100%的数据点，n,m <= 100000，∑a[i] <= 10^9

解题思路：

先考虑一个集合的神秘数是如何确定的。
将集合中的数从小到大排序，若当前数为x，如果其前面的数组成的神秘数为ans，若ans>=x，那当前神秘数就是ans+x+1，否则答案就是ans。
所以可以得到以下算法：初始ans为1，每次求小于ans的数之和sum，令ans=sum+1，直到sum小于ans即为答案。
不难发现每次ans至少会翻倍，所以只需查询logn次。
区间查询可以用主席树实现。

#include<bits/stdc++.h>
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=100005,M=5000005,INF=1e9;
struct node{int lc,rc,sum;}tr[M];
int n,m,tot,rt[N];

void Insert(int y,int &x,int l,int r,int v)
{
    tr[x=++tot]=tr[y],tr[x].sum+=v;
    if(l==r)return;
    int mid=l+r>>1;
    if(v<=mid)Insert(tr[y].lc,tr[x].lc,l,mid,v);
    else Insert(tr[y].rc,tr[x].rc,mid+1,r,v);
}

int query(int x,int y,int l,int r,int lsum,int v)
{
    if(l==r)return lsum+tr[y].sum-tr[x].sum;
    int mid=l+r>>1;
    if(v<=mid)return query(tr[x].lc,tr[y].lc,l,mid,lsum,v);
    else return query(tr[x].rc,tr[y].rc,mid+1,r,lsum+tr[tr[y].lc].sum-tr[tr[x].lc].sum,v);
}

int main()
{
    //freopen("lx.in","r",stdin);
    n=getint();
    for(int i=1;i<=n;i++)Insert(rt[i-1],rt[i],1,INF,getint());
    m=getint();
    while(m--)
    {
        int l=getint(),r=getint(),ans=1;
        while(1)
        {
            int tmp=query(rt[l-1],rt[r],1,INF,0,ans);
            if(tmp>=ans)ans=tmp+1;
            else break;
        }
        printf("%d\n",ans);
    }
}