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;
} 

你可能感兴趣的:(可持久化数据结构,BZOJ)