day1 晚上 P4145 上帝造题的七分钟2 / 花神游历各国 线段树

#include<iostream>
#include<cstdio>
#include<cmath>

using namespace std;

const int maxn=1e6+10;
struct node
{
    int l,r;
    long long sum,maxx;
}tree[maxn];
long long n;
long long num[maxn];
long long pushup(long long x)
{
    tree[x].sum=tree[x<<1].sum+tree[x<<1|1].sum;
    if(tree[x<<1].maxx>tree[x<<1|1].maxx)
    {
        tree[x].maxx=tree[x<<1].maxx;
    }
    else
    {
        tree[x].maxx=tree[x<<1|1].maxx;
    }
}
void build(long long x,long long l,long long r)
{
    tree[x].l=l;
    tree[x].r=r;
    if(l==r)
    {
        tree[x].sum=tree[x].maxx=num[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    pushup(x);
}
void modify(long long x,long long l,long long r)
{
    if(tree[x].l==tree[x].r)
    {
        tree[x].sum=sqrt(tree[x].sum);
        tree[x].maxx=sqrt(tree[x].maxx);
        return ;
    }
    long long mid=(tree[x].l+tree[x].r)>>1;
    if(l<=mid&&tree[x<<1].maxx>1)
    {
        modify(x<<1,l,r);
    }
    if(r>mid&&tree[x<<1|1].maxx>1)
    {
        modify(x<<1|1,l,r);
    }
    pushup(x);
}
long long query(long x,long long l,long long r)
{
    if(l<=tree[x].l&&r>=tree[x].r)
    {
        return tree[x].sum;
    }
    long long  mid=(tree[x].l+tree[x].r)>>1;
    long long  ans=0;
    if(l<=mid)
    {
        ans+=query(x<<1,l,r);
    }
    if(r>mid)
    {
        ans+=query(x<<1|1,l,r);
    }
    return ans;
}
int main()
{
    cin>>n;
    for(long long i=1;i<=n;i++)
    {
        cin>>num[i];
    }
    build(1,1,n);
    long long m,opt,l,r;
    cin>>m;
    while(m--)
    {
        cin>>opt>>l>>r;
        if(l>r)
        {
            swap(l,r);
        }
        if(opt==0)
        {
            modify(1,l,r);
        }
        else
        {
            cout<<query(1,l,r)<<endl;
        }
    }
    return 0;
}