BZOJ 4408: [Fjoi 2016]神秘数（可持久化线段树）

题目描述

http://www.lydsy.com/JudgeOnline/problem.php?id=4408

题目大意：求最小不能被一段区间中某些数的和表示的数。（还是看题面吧）

---

思路

可持久化线段树好题。话说为什么我打可持久总会看到可吃鸡。。

首先设能表示的一段区间为[1,x]，则神秘数为x+1。当前加入一个数y，分为两种情况：

①y<=x+1，则能表示的区间变成[1,x+y]，神秘数变成x+y+1。

②y>x+1，则x+1不能被表示，能表示的区间不变，神秘数不变。

于是我们维护区间的和。设神秘数为ans，一开始ans=1。找到一个区间中小于等于ans的数并求和，记为Get。如果Get>=ans， 则区间可以扩展变成[1,Get]，ans=Get+1。

如果Get < ans，代表此时ans不会改变，神秘数就是现在的ans。

该题神奇的地方就在y>x+1的y对神秘数不产生任何影响，喵处就是y<=x+1的y可以一次性更新答案。

而找一个区间中小于等于某数的和可以用可持久化线段树来维护。即每个点开权值线段树上的一条链，然后维护前缀和就行了。

由于每次询问时间logN，而每次ans至少翻一番，所以总时间就是O(nlogN+mlogNlogN)。

---

代码

#include 
#define maxn 100010
#define Lg 32

using namespace std;

int a[maxn], n, m, cnt, ToT, ans, Get;

struct Tnode{
    Tnode *lson, *rson;
    int sum;
}tree[maxn*Lg], *Root[maxn];

Tnode *NewTnode(){
    tree[cnt].lson = tree[cnt].rson = tree;
    tree[cnt].sum = 0;
    return tree+cnt++;
}

void Update(Tnode *root, int L, int R, int x){
    if(L == R){
        root->sum += x;
        return;
    }

    int mid = (L + R) >> 1;
    Tnode *p = NewTnode();

    if(x <= mid){
        *p = *(root->lson);
         root->lson = p;
         Update(p, L, mid, x);
    }
    else{
        *p = *(root->rson);
        root->rson = p;
        Update(p, mid+1, R, x);
    }

    root->sum = root->lson->sum + root->rson->sum;
}

int Query(Tnode *root1, Tnode *root2, int L, int R, int x){
    if(L == R)  return root2->sum - root1->sum;

    int mid = (L + R) >> 1;
    if(x <= mid)  return Query(root1->lson, root2->lson, L, mid, x);
    else  return root2->lson->sum - root1->lson->sum + Query(root1->rson, root2->rson, mid+1, R, x);
}

int main(){

    scanf("%d", &n);
    for(int i = 1; i <= n; i++)  scanf("%d", &a[i]), ToT += a[i];

    Root[0] = NewTnode();
    for(int i = 1; i <= n; i++){
        Root[i] = NewTnode();
        *Root[i] = *Root[i-1];
        Update(Root[i], 1, ToT, a[i]);
    }

    scanf("%d", &m);

    int l, r;
    for(int i = 1; i <= m; i++){
        scanf("%d%d", &l, &r);
        ans = 1;
        for(;;){
            Get = Query(Root[l-1], Root[r], 1, ToT, ans);
            if(Get < ans)  break;
            ans = Get + 1;
        }
        printf("%d\n", ans);
    }

    return 0;
} 

---