loj #6281. 數列分塊入門 5

#6281. 數列分塊入門 5

題目描述

給出一個長爲 nnn 的數列 a1…ana_1\ldots a_na​1​​…a​n​​，以及 nnn 個操做，操做涉及區間開方，區間求和。

輸入格式

第一行輸入一個數字 nnn。

第二行輸入 nnn 個數字，第 i 個數字爲 aia_ia​i​​，以空格隔開。

接下來輸入 nnn 行詢問，每行輸入四個數字 opt,l,r,c\mathrm{opt}, l, r, copt,l,r,c，以空格隔開。

若 opt=0\mathrm{opt} = 0opt=0，表示將位於 [l,r][l, r][l,r] 的之間的數字都開方。對於區間中每一個 ${\mathrm{ai(l\le i\le r),ai\leftarrow \lfloor \surd ai\rfloor }}^{}$

若 opt=1\mathrm{opt} = 1opt=1，表示詢問位於 [l,r][l, r][l,r] 的全部數字的和。

輸出格式

對於每次詢問，輸出一行一個數字表示答案。

樣例

樣例輸入

4
1 2 2 3
0 1 3 1
1 1 4 4
0 1 2 2
1 1 2 4

樣例輸出

6
2

數據範圍與提示

對於 100% 100\%100% 的數據，1≤n≤50000,−231≤others 1 \leq n \leq 50000, -2^{31} \leq \mathrm{others}1≤n≤50000,−2​31​​≤others、ans≤231−1 \mathrm{ans} \leq 2^{31}-1ans≤2​31​​−1。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 100010
using namespace std;
long long n,m,a[maxn],sum[maxn],block,pos[maxn];
bool v[maxn];
int findl(int id){return (id-1)*block+1;}
int findr(int id){return min(id*block,n);}
long long qread(){
    long long i=0,j=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')j=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){i=i*10+ch-'0';ch=getchar();}
    return i*j;
}
void reset(int id){
    if(v[id])return;
    v[id]=1;sum[id]=0;
    for(int i=findl(id);i<=findr(id);i++){
        a[i]=sqrt(a[i]);
        sum[id]+=a[i];
        if(a[i]>1)v[id]=0;
    }
}
void update(int l,int r){
    if(!v[pos[l]]){
        for(int i=l;i<=min(findr(pos[l]),r);i++){
            sum[pos[l]]-=a[i];
            a[i]=sqrt(a[i]);
            sum[pos[l]]+=a[i];
        }
        v[pos[l]]=1;
        for(int i=findl(pos[l]);i<=findr(pos[l]);i++)
            if(a[i]>1)v[pos[l]]=0;
    }
    if(pos[l]==pos[r])return;
    if(!v[pos[r]]){
        for(int i=findl(pos[r]);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=findl(pos[r]);i<=findr(pos[r]);i++)
            if(a[i]>0)v[pos[r]]=0;
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)reset(i);
}
void query(int l,int r){
    long long res=0;
    for(int i=l;i<=min(findr(pos[l]),r);i++)res+=a[i];
    if(pos[l]!=pos[r])
        for(int i=findl(pos[r]);i<=r;i++)res+=a[i];
    for(int i=pos[l]+1;i<=pos[r]-1;i++)res+=sum[i];
    cout<<res<<endl;
}
int main(){
    n=qread();
    for(int i=1;i<=n;i++)a[i]=qread();
    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++){
        long long op=qread(),l=qread(),r=qread(),c=qread();
        if(op==0)update(l,r);
        else query(l,r);
    }
    return 0;
}