【 bzoj 4299 】 Codechef FRBSUM - 主席树乱搞

  一眼看status吓得我还以为是分块。。。其实是水题。
  考虑给定一个集合,如何得到FRBSUM。
  按某种顺序从集合中取数,设之前取的数可以取到的最大值为 mx (雾)且 [1,mx] 都可以取到,然后现在新加了一个数 x ,如果 xmx+1 ,那么最大值可以由mx扩到 mx+lim ,同时 (x,mx+lim) 内的数显然也可以得到。而如果 x>mx+1 时,取FRBSUM =mx+1 即可。如果这个顺序是从小到大,那么就会有神奇的事情发生(雾)。
  至此我们可以得到一个算法,设当前查询集合为S,先从1开始,判断S内小于1的数的和,然后再用这个和+1扔进S内查询一下,直至查询得到的结果不变。
  于是就可以很愉快地用主席树乱搞了。
  由于每次倍增最多增加一倍,于是查询次数是 O(logn) 的,每次查询的时间复杂度是 O(logn) 。总时间复杂度是 O(nlog2n) ,空间复杂度 O(nlogSmax)
  

#include <bits/stdc++.h>
#define rep(i,a,b) for(int i = a , _ = b ; i <= _ ; i ++)
#define per(i,a,b) for(int i = a , _ = b ; i >= _ ; i --)
#define maxn 100007
#define maxs 4000007

inline int rd() {
    char c = getchar();
    while (!isdigit(c)) c = getchar() ; int x = c - '0';
    while (isdigit(c = getchar())) x = x * 10 + c - '0';
    return x;
}

inline void upmax(int&a , int b) { if (a < b) a = b ; }

typedef int arr[maxn];
typedef int seg[maxs];

arr rt , a;
seg lc , rc , sum;
int tot , n , m , lim;

void update(int pr , int&nr , int l , int r , int p) {
    if (!nr) nr = ++ tot;
    sum[nr] = sum[pr] + p;
    if (l == r) return;
    int m = l + r >> 1;
    if (p <= m)
        rc[nr] = rc[pr] , update(lc[pr] , lc[nr] , l , m , p);
    else
        lc[nr] = lc[pr] , update(rc[pr] , rc[nr] , m + 1 , r , p);
}

void input() {
    n = rd();
    rep (i , 1 , n) a[i] = rd() , upmax(lim , a[i]);
    rep (i , 1 , n) update(rt[i - 1] , rt[i] , 1 , lim , a[i]);
}

int query(int pr , int nr , int l , int r , int p) {
    if (l > p) return 0;
    if (!(sum[nr] - sum[pr])) return 0;
    if (r <= p) return sum[nr] - sum[pr];
    int m = l + r >> 1;
    int ret = query(lc[pr] , lc[nr] , l , m , p);
    if (p > m) ret += query(rc[pr] , rc[nr] , m + 1 , r , p);
    return ret;
}

inline int query(int l , int r) {
    for (int t = 1 , x;;t = x + 1)
        if ((x = query(rt[l] , rt[r] , 1 , lim , t)) < t)
            return t;
}

void solve() {
    m = rd();
    rep (i , 1 , m) {
        int l = rd() , r = rd();
        int ans = query(l - 1 , r);
        printf("%d\n" , ans);
    }
}

int main() {
    #ifndef ONLINE_JUDGE
        freopen("data.txt" , "r" , stdin);
    #endif
    input();
    solve();
    return 0;
}

你可能感兴趣的:(【 bzoj 4299 】 Codechef FRBSUM - 主席树乱搞)