判断相同区间(lazy) 多校8 HDU 5828 Rikka with Sequence

// 判断相同区间(lazy) 多校8 HDU 5828 Rikka with Sequence
// 题意：三种操作，1增加值，2开根，3求和
// 思路：这题与HDU 4027 和HDU 5634 差不多
// 注意开根号的话，遇到极差等于1的，开根号以后有可能还是差1.如
// 2 3 2 3。。。
// 8 9 8 9。。。
// 2 3 2 3。。。
// 8 9 8 9。。。
// 剩下就是遇到区间相等的话，就直接开根号不往下传


#include <bits/stdc++.h>
using namespace std;
#define LL long long
const int inf = 0x3f3f3f3f;
const int MOD =998244353; 
const int N =1e5+10; 
#define clc(a,b) memset(a,b,sizeof(a))
const double eps = 1e-7;
void fre(){freopen("in.txt","r",stdin);}
void freout() {freopen("out.txt","w",stdout);}
inline int read() {int x=0,f=1;char ch=getchar();while(ch>'9'||ch<'0') {if(ch=='-') f=-1;ch=getchar();}while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();}return x*f;}
int n,q;
int a[N];
struct node{
    int l,r,len;
    LL sum,lazy;
    LL mx,mn;
}t[N<<2];

void pushup(int rt){
    t[rt].sum=t[rt<<1].sum+t[rt<<1|1].sum;
    t[rt].mx=max(t[rt<<1].mx,t[rt<<1|1].mx);
    t[rt].mn=min(t[rt<<1|1].mn,t[rt<<1].mn);
}
void pushdown(int rt){
    if(t[rt].lazy){
        t[rt<<1].lazy+=t[rt].lazy;
        t[rt<<1].mx+=t[rt].lazy;
        t[rt<<1].mn+=t[rt].lazy;
        t[rt<<1].sum+=t[rt<<1].len*t[rt].lazy;
        t[rt<<1|1].lazy+=t[rt].lazy;
        t[rt<<1|1].mx+=t[rt].lazy;
        t[rt<<1|1].mn+=t[rt].lazy;
        t[rt<<1|1].sum+=t[rt<<1|1].len*t[rt].lazy;
        t[rt].lazy=0;
    }
}
void build(int rt,int l,int r){
    t[rt].l=l;
    t[rt].r=r;
    t[rt].lazy=0;
    t[rt].mx=0;
    t[rt].mn=inf;
    t[rt].len=r-l+1;
    if(l==r){
       t[rt].sum=a[l],t[rt].mx=t[rt].mn=a[l];
       return;    
    }
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    pushup(rt);
}

void update1(int rt,int l,int r,int z){
    if(t[rt].l>=l&&t[rt].r<=r){
        t[rt].sum+=(LL)t[rt].len*z;
        t[rt].lazy+=z;
        t[rt].mn+=z;
        t[rt].mx+=z;
        return;
    }
    pushdown(rt);
    int mid=(t[rt].l+t[rt].r)>>1;
    if(r<=mid) update1(rt<<1,l,r,z);
    else if(l>mid) update1(rt<<1|1,l,r,z);
    else {
        update1(rt<<1,l,mid,z);
        update1(rt<<1|1,mid+1,r,z);
    }
    pushup(rt);
}

void update2(int rt,int l,int r){
    if(t[rt].l>=l&&t[rt].r<=r){
        if(t[rt].mx==t[rt].mn){
           LL tem=(LL)sqrt(t[rt].mx);
           t[rt].sum=(LL)tem*t[rt].len;
           t[rt].lazy-=(t[rt].mx-tem);
           t[rt].mx=t[rt].mn=tem;
           return;
        }
        else if(t[rt].mx==t[rt].mn+1){
            LL tem1=(LL)sqrt(t[rt].mx);
            LL tem2=(LL)sqrt(t[rt].mn);
            if(tem1==tem2+1){
                t[rt].sum-=(LL)(t[rt].mx-tem1)*t[rt].len;
                t[rt].lazy-=(t[rt].mx-tem1);
                t[rt].mx=tem1;
                t[rt].mn=tem2;
                return;
            }
        }
    }
    pushdown(rt);
    int mid=(t[rt].l+t[rt].r)>>1;
    if(r<=mid) update2(rt<<1,l,r);
    else if(l>mid) update2(rt<<1|1,l,r);
    else {
        update2(rt<<1,l,mid);
        update2(rt<<1|1,mid+1,r);
    }
    pushup(rt);
}

LL query(int rt,int l,int r){
    if(t[rt].l>=l&&t[rt].r<=r){
        return t[rt].sum;
    }
    pushdown(rt);
    int mid=(t[rt].l+t[rt].r)>>1;
    if(r<=mid) return query(rt<<1,l,r);
    else if(l>mid) return query(rt<<1|1,l,r);
    else {
        return query(rt<<1,l,mid)+query(rt<<1|1,mid+1,r);
    }
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
       scanf("%d%d",&n,&q);
       for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
       }
       build(1,1,n);
       while(q--){
           int op,x,y,z;
           scanf("%d%d%d",&op,&x,&y);
           if(op==1){
               scanf("%d",&z);
               update1(1,x,y,z);
           }
           else if(op==2){
               update2(1,x,y);
           } 
           else {
               LL ans=query(1,x,y);
               printf("%I64d
",ans);
           }
       }
    }
    return 0;
}