BZOJ3211 花神游历各国

Description

 

Input

 

Output

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

 

Sample Input

4

1 100 5 5

5

1 1 2

2 1 2

1 1 2

2 2 3

1 1 4

Sample Output

101

11

11

HINT

 

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

 

每个数最多log(logAi)次就会变成0或1，所以每次暴力修改一个单点可行。

但我们怎么找到区间中所有不是0和1的点呢？用并查集维护每个位置下一个合法的位置，细节见代码。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
#include<cmath>
#define rep(s,t) for(int i=s;i<=t;i++)
#define ren for(int i=first[x];i!=-1;i=next[i])
using namespace std;
typedef long long ll;
inline ll read() {
    ll x=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=100010;
int n,fa[maxn],A[maxn];
inline int findset(int x) {return !fa[x]||x==fa[x]?x:fa[x]=findset(fa[x]);}
ll c[maxn];
void add(int x,ll v) {for(;x<=n;x+=x&-x) c[x]+=v;}
ll sum(int x) {ll ret=0;for(;x;x-=x&-x) ret+=c[x];return ret;}
int main() {
    n=read();
    rep(1,n) {
        add(i,A[i]=read());
        if(A[i]<=1) fa[i]=i+1;
    }
    int q=read();
    while(q--) {
        int t=read(),l=read(),r=read();
        if(t==1) printf("%lld\n",sum(r)-sum(l-1));
        else {
            for(int i=findset(l);i<=r;i=findset(i+1)) {
                int t=sqrt(A[i]);
                add(i,t-A[i]);A[i]=t;
                if(A[i]<=1) fa[i]=findset(i+1);
            }
        }
    }
    return 0;
}