3211: 花神遊歷各國

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

可以註意到題目中給出的最大數1e9,在開根幾次之後都變成了0或1,0或1開根後不變
也就是說對於每個位置,修改操作次數是有限的,修改次數達到上限以後就不用修改了

開lazy標記當前區間是否開根開到全是0或1,sum表示區間和
考慮到修改不超過10次,我們每次修改都深入到最底層節點暴力修改再傳給父親信息即可
當一個節點的兒子全被lazy標記,它也可以打上lazy標記

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#define ll long long
#define N 100005
using namespace std;
int n,m,a[N],lz[N<<2
 
];ll sum[N<<2];
void pushup(int u){
    int l=u<<1,r=l|1;
    sum[u]=sum[l]+sum[r];
    lz[u]=lz[l]&lz[r];
}
void build(int u,int l,int r){
    if(l==r){
        sum[u]=a[l];
        if(sum[u]<=1)lz[u]=1;
        return;
    }
    int mid=l+r>>1;
    build(u<<1
 
,l,mid);
    build(u<<1|1,mid+1,r);
    pushup(u);
}
void update(int u,int L,int R,int l,int r){
    if(lz[u])return;
    if(L==R){
        sum[u]=sqrt(sum[u]);
        if(sum[u]<=1)lz[u]=1;
        return;
    }
    int mid=L+R>>1;
    if(l<=mid)update(u<<1,L,mid,l,r);
    if(r>mid)update(u<<1|1,mid+1,R,l,r);
    pushup(u);
}
ll query(int u,int L,int R,int l,int r){
    if(l<=L&&R<=r)return sum[u];
    int mid=L+R>>1;
    ll t=0;
    if(l<=mid)t+=query(u<<1,L,mid,l,r);
    if(r>mid)t+=query(u<<1|1,mid+1,R,l,r);
    return t;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    while(m--){
        int op,l,r;
        scanf("%d%d%d",&op,&l,&r);
        if(l>r)swap(l,r);
        if(op==1)printf("%lld\n",query(1,1,n,l,r));
        else update(1,1,n,l,r);
    }
    return 0;
}

bzoj3211花神遊歷各國 線段樹