BZOJ3211 花神游历各国 【树状数组 + 并查集】

题目

输入格式

输出格式

每次x=1时，每行一个整数，表示这次旅行的开心度

输入样例

4

1 100 5 5

5

1 1 2

2 1 2

1 1 2

2 2 3

1 1 4

输出样例

101

11

11

数据

对于100%的数据， n ≤ 100000，m≤200000 ,data[i]非负且小于10^9

题解

类似于重复开根以及取模之类的操作都有个共性，就是重复取的次数非常少
以本题开根为例，当取到一定次数时，就会变为1
我们用树状数组维护区间和
用并查集维护当前位置往下【包括当前位置】，最近还可以开根的位置
每次暴力开根维护树状数组即可【最多开5*N次】

#include<iostream>
#include<cstdio>
#include<cmath>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define lbt(x) (x & -x)
using namespace std;
const int maxn = 100005,maxm = 100005,INF = 1000000000;
inline int RD(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}
    while (c >= 48 && c <= 57) {out = (out << 1) + (out << 3) + c - '0'; c = getchar();}
    return out * flag;
}
int N,M,pre[maxn],num[maxn];
LL A[maxn];
void add(int u,LL v){while (u <= N) A[u] += v,u += lbt(u);}
LL query(int u){LL ans = 0; while(u) ans += A[u],u -= lbt(u); return ans;}
LL sum(int l,int r){return query(r) - query(l - 1);}
int find(int u){return u == pre[u] ? u : pre[u] = find(pre[u]);}
int main(){
    N = RD();
    REP(i,N){
        num[i] = RD(); add(i,num[i]);
        pre[i] = (num[i] != 0 && num[i] != 1) ? i : i + 1;
    }pre[N + 1] = N + 1;
    M = RD();
    int cmd,l,r,u,v;
    while (M--){
        cmd = RD(); l = RD(); r = RD();
        if (cmd & 1) printf("%lld\n",sum(l,r));
        else {
            u = find(l);
            while (u <= r){
                v = (LL)sqrt(num[u]); add(u,v - num[u]);
                num[u] = v;
                if (num[u] == 1 || num[u] == 0) pre[u] = u + 1;
                u++;
            }
        }
    }
    return 0;
}