【题解】洛谷P4145 花神游历各国（线段树）

洛谷P4145：https://www.luogu.org/problemnew/show/P4145

思路

这道题的重点在于sqrt(1)=1 一个限制条件

与正常线段树不同的是区间修改为开方

那么我们用一个数组记录每个区间的最大值 只有当这个区间的最大值大于1时才需要开方

因此 当我们更新到叶子节点时把每个区间的最大值和sum值开方即可

注意题目中说l可能大于r 要交换

代码

#include<iostream>
#include<cmath>
using namespace std;
#define ll long long
#define maxn 100010
ll sum[maxn<<],Max[maxn<<],a[maxn];
ll n,m;
void build(ll l,ll r,ll k)
{
    if(l==r)
    {
        sum[k]=a[l];
        Max[k]=sum[k];//叶子节点
        return;
    }
    ll mid=(l+r)>>;
    build(l,mid,k<<);
    build(mid+,r,k<<|);
    sum[k]=sum[k<<]+sum[k<<|];
    Max[k]=max(Max[k<<],Max[k<<|]);//更新上层的值
    return;
}
ll query(ll x,ll y,ll l,ll r,ll k)//常规询问
{
    if(x<=l&&r<=y) return sum[k];
    ll res=;
    ll mid=(l+r)>>;
    if(x<=mid) res+=query(x,y,l,mid,k<<);
    if(y>mid) res+=query(x,y,mid+,r,k<<|);
    return res;
}
void update(ll x,ll y,ll l,ll r,ll k)
{
    if(l==r)//叶子节点时
    {
        sum[k]=sqrt(sum[k]);
        Max[k]=sqrt(Max[k]);
        return;
    }
    ll mid=(l+r)>>;
    if(x<=mid&&Max[k<<]>) update(x,y,l,mid,k<<);//满足最大值大于1
    if(y>mid&&Max[k<<|]>) update(x,y,mid+,r,k<<|);
    sum[k]=sum[k<<]+sum[k<<|];//更新上层的值
    Max[k]=max(Max[k<<],Max[k<<|]);
}
int main()
{
    cin>>n;
    for(ll i=;i<=n;i++) cin>>a[i];
    build(,n,);
    cin>>m;
    for(ll i=;i<=m;i++)
    {
        ll k,l,r;
        cin>>k>>l>>r;
        if(l>r)//交换
        {
            int temp;
            temp=l;
            l=r;
            r=temp;
        }
        if(k==)
        {
            update(l,r,,n,);
        }
        if(k==)
        {
            cout<<query(l,r,,n,)<<endl;
        }
    }
}