loj#6281. 数列分块入门 5

#6281. 数列分块入门 5

题目：传送门 

简要题意：

　　 给出一个长为 n 的数列，以及 n 个操作，操作涉及区间开方，区间求和。

---

题解：

　　 怎么说...这道题loj的数据有点水。

　　 和bzoj花神游历各国是一样的...但是loj没有卡掉不完全优化的代码。

　　 基础操作就不说了（同分块4），主要讲优化吧：

　　 不难发现，因为开方后我们是向下取整的，那么如果一直开方下去，整个序列一定会变成0或1。那么如果当前区间都是0或1就没有操作的必要了嘛。

　　 讲到这里思路就已经很清晰了，但是一开始的时候本蒟蒻并没有优化头尾的块，依然在loj AC了...bzoj就挂了，事实证明：暴力加优化才是分块真正的难点！

---

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long LL;
LL n,m,a[110000],sum[110000];
LL block,pos[110000];
bool v[110000];
void reset(LL x)
{
    if(v[x])return ;
    v[x]=1;sum[x]=0;
    for(int i=(x-1)*block+1;i<=x*block;i++)
    {
        a[i]=sqrt(a[i]);sum[x]+=a[i];
        if(a[i]>1)v[x]=0;
    }
}
void update(LL l,LL r)
{
    if(!v[pos[l]])
    {
        for(int i=l;i<=min(pos[l]*block,r);i++)
        {
            sum[pos[l]]-=a[i];
            a[i]=sqrt(a[i]);
            sum[pos[l]]+=a[i];
        }
        v[pos[l]]=1;for(int i=(pos[l]-1)*block+1;i<=pos[l]*block;i++)if(a[i]>1)v[pos[l]]=0;
    }
    if(pos[l]!=pos[r])
    {
        if(!v[pos[r]])
        {
            for(int i=(pos[r]-1)*block+1;i<=r;i++)
            {
                sum[pos[r]]-=a[i];
                a[i]=sqrt(a[i]);
                sum[pos[r]]+=a[i];
            }
            v[pos[r]]=1;for(int i=(pos[r]-1)*block;i<=pos[r]*block;i++)if(a[i]>1)v[pos[r]]=0;
        }
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)reset(i);
}
void sol(LL l,LL r)
{
    LL ans=0;
    for(int i=l;i<=min(pos[l]*block,r);i++)ans+=a[i];
    if(pos[l]!=pos[r])
        for(int i=(pos[r]-1)*block+1;i<=r;i++)
            ans+=a[i];
    for(int i=pos[l]+1;i<=pos[r]-1;i++)ans+=sum[i];
    printf("%lld
",ans);
}
int main()
{
    memset(v,false,sizeof(v));
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
    block=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        pos[i]=(i-1)/block+1;
        sum[pos[i]]+=a[i];
    }
    for(int i=1;i<=n;i++)
    {
        LL opt,l,r,c;scanf("%lld%lld%lld%lld",&opt,&l,&r,&c);
        if(opt==0)update(l,r);
        else sol(l,r);
    }
    return 0;
}