数列分块入门 5

题目链接
数据最大范围是2^31,所以最多开5次方就会话成<=1的数。用一个数组进行标记块内的元素是不是全部都小于等于1。

#include<stdio.h>
#include<cmath>
#include<set>
#include<vector>
#include<algorithm>
#include<iostream>
#include<string.h>
#include<string>
using namespace std;
typedef long long int ll;
const int N=50000+10;
#define int long long
int v[N],pos[N],flag[N];
int sum[N],n,m;
void solve_sqrt(int x)
{
    if(flag[x])return ;
    flag[x]=1;
    sum[x]=0;
    for(int i=(x-1)*m+1;i<=min(x*m,n);i++)
    {
        v[i]=sqrt(v[i]); sum[x]+=v[i];
        if(v[i]>1)flag[x]=0;
    }
}
void Sq(int l,int r)
{
    for(int i=l;i<=min(pos[l]*m,r);i++)
    {
        sum[pos[l]]-=v[i];
        v[i]=sqrt(v[i]);
        sum[pos[l]]+=v[i];
    }
    if(pos[l]!=pos[r])
    {
        for(int i=(pos[r]-1)*m+1;i<=r;i++)
        {
            sum[pos[r]]-=v[i];
            v[i]=sqrt(v[i]);
            sum[pos[r]]+=v[i];
        }
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
        solve_sqrt(i);
}
int query(int l,int r)
{
    int ans=0;
    for(int i=l;i<=min(pos[l]*m,r);i++)
    {
        ans+=v[i];
    }
    if(pos[l]!=pos[r])
    {
        for(int i=(pos[r]-1)*m+1;i<=r;i++)
        ans+=v[i];
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
        ans+=sum[i];
    return ans;
}
signed main()
{
    scanf("%lld",&n);
    m=sqrt(n);
    for(int i=1;i<=n;i++)scanf("%lld",v+i);
    for(int i=1;i<=n;i++)
    {
        pos[i]=(i-1)/m+1;
        sum[pos[i]]+=v[i];
    }
    for(int i=1;i<=n;i++)
    {
        int l,r,f,c;
        scanf("%lld%lld%lld%lld",&f,&l,&r,&c);
        if(f==0)Sq(l,r);
        if(f==1)printf("%lld\n",query(l,r));
    }
    return 0;
}