LibreOJ 6281 数列分块入门5

题目链接：https://loj.ac/problem/6281

参考博客：https://blog.csdn.net/qq_36038511/article/details/79725027

我一开始看到开方有点懵，看了上面的博客说2^31差不多5次开方就快变成1了，可以开标记数组，于是按照这个思路写了下，然后就过了，看来以后还要多看题意分析数据。

代码：

#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdio>
using namespace std;
typedef long long LL;
#define eps 1e-8
#define INF 0x3f3f3f3f
#define maxn 50010
int n,m,k,t,block;
int a[maxn],tag[maxn],sum[maxn],vis[maxn],lump[maxn];
void kaif(int l,int r)//把l到r之间的数字开方
{
    for(int i=l;i<=min(lump[l]*block,r);i++)//直接改左边的不完整区间，同时该区间和
    {
        sum[lump[l]]-=a[i];
        a[i]=sqrt(a[i]);
        sum[lump[l]]+=a[i];
    }
    if(lump[l]!=lump[r])
    {
        for(int i=(lump[r]-)*block+;i<=r;i++)
        {
            sum[lump[r]]-=a[i];
            a[i]=sqrt(a[i]);
            sum[lump[r]]+=a[i];
        }
    }
    for(int i=lump[l]+;i<=lump[r]-;i++)//记录这个区间开方了多少次
    {
        tag[i]++;
    }
}
int cal(int x,int num)//求数字x开方num次得出的结果
{
    while(num--&&x>)
    {
        x=sqrt(x);
    }
    return x;
}
int get_sum(int l,int r)//求区间和
{
    int ans=;
    for(int i=l;i<=min(lump[l]*block,r);i++)//左区间暴力求和
    {
        ans+=cal(a[i],tag[lump[l]]);
    }
    if(lump[l]!=lump[r])
    {
        for(int i=(lump[r]-)*block+;i<=r;i++)//右区间
        {
            ans+=cal(a[i],tag[lump[r]]);
        }
    }
    for(int i=lump[l]+;i<=lump[r]-;i++)
    {
        if(!vis[i])//如果vis[i]==0,表示区间i还有大于1的数字
        {
            int flag=;//标记看开方之后是否还有大于1的数字
            sum[i]=;//区间和清空
            for(int j=(i-)*block+;j<=i*block;j++)
            {
                a[j]=cal(a[j],tag[i]);
                sum[i]+=a[j];//重新计算区间和
                if(a[j]>)//只要有一个大于1
                flag=;
            }
            tag[i]=;//区间未开方数字清零
            if(!flag)
            vis[i]=;
        }
        ans+=sum[i];
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    block=sqrt(n);
    fill(sum,sum+maxn-,);
    fill(tag,tag+maxn-,);
    fill(vis,vis+maxn-,);
    for(int i=;i<=n;i++)
    {
        scanf("%d",&a[i]);
        lump[i]=(i-)/block+;
        sum[lump[i]]+=a[i];
    }
    for(int i=;i<=n;i++)
    {
        int op,l,r,w;
        scanf("%d%d%d%d",&op,&l,&r,&w);
        if(op==)
        kaif(l,r);
        else
        printf("%d\n",get_sum(l,r));
    }
    return ;
}