【洛谷】P4145 上帝造题的七分钟 2 / 花神游历各国 (线段树)

P4145 上帝造题的七分钟 2 / 花神游历各国

最大数为\(10^{12}\)，开方最多开\(6\)次，线段树维护一个区间最大值，如果最大值为\(1\)就不需要进行操作了，否则暴力递归.

#include <bits/stdc++.h>

using namespace std;
const int N=1e6+10;
const int mod=1e9+7;
#define ll long long
#define pb push_back

int n,m;
ll a[N];

struct Node{
    int l,r;
	ll mx;
    ll sum;
}tr[N<<4];

void push_up(int u){
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
	tr[u].mx=max(tr[u<<1].mx,tr[u<<1|1].mx);
}

void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,r,a[l],a[l]};
        return;
    }
    tr[u]={l,r,0,0};
    int mid=(l+r)>>1;
    build(u<<1,l,mid);
    build(u<<1|1,mid+1,r);
    push_up(u);
}

void update(int u,int L,int R){
    if(tr[u].l==tr[u].r){
        tr[u].sum=sqrt(tr[u].sum);
		tr[u].mx=sqrt(tr[u].mx);
        return;
    }
    int mid=(tr[u].l+tr[u].r)>>1;
    if(L<=mid && tr[u<<1].mx>1) update(u<<1,L,R);
    if(R>mid && tr[u<<1|1].mx>1) update(u<<1|1,L,R);
    push_up(u);
}

ll query(int u,int L,int R){
    if(tr[u].l>=L && tr[u].r<=R){
        return tr[u].sum;
    }
    int mid=(tr[u].l+tr[u].r)>>1;
    ll res=0;
    if(L<=mid) res+=query(u<<1,L,R);
    if(R>mid) res+=query(u<<1|1,L,R);
    return res;
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%lld",&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==0){
            update(1,l,r);
        }
        else printf("%lld\n",query(1,l,r));
    }

    return 0;
}