Loj 6281. 数列分块入门 5

题目描述

给出一个长为 nnn 的数列 a1…ana_1\ldots a_na1…an，以及 nnn 个操作，操作涉及区间开方，区间求和。

输入格式

第一行输入一个数字 nnn。

第二行输入 nnn 个数字，第 i 个数字为 aia_iai，以空格隔开。

接下来输入 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] 的之间的数字都开方。对于区间中每个 ai(l≤i≤r), ai←⌊ai⌋a_i(l\le i\le r),\: a_i ← \left\lfloor \sqrt{a_i}\right\rfloorai(l≤i≤r),ai←⌊ai⌋

若 opt=1\mathrm{opt} = 1opt=1，表示询问位于 [l,r][l, r][l,r] 的所有数字的和。

输出格式

对于每次询问，输出一行一个数字表示答案。

样例

样例输入

样例输出

6
2

数据范围与提示

对于 100% 100\%100% 的数据，1≤n≤50000,−231≤others 1 \leq n \leq 50000, -2^{31} \leq \mathrm{others}1≤n≤50000,−231≤others、ans≤231−1 \mathrm{ans} \leq 2^{31}-1ans≤231−1。

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3
  4 #define F(i,a,b) for(int i=a;i<=b;i++)
  5 #define D(i,a,b) for(int i=a;i>=b;i--)
  6 #define ms(i,a)  memset(a,i,sizeof(a))
  7 #define LL       long long
  8 #define st(x)    ((x-1)*B+1)
  9 #define ed(x)    min(x*B,n)
 10 #define bl(x)    ((x-1)/B+1)
 11
 12 int inline read(){
 13     int x=0,w=0; char c=getchar();
 14     while (c<'0' || c>'9') w+=c=='-',c=getchar();
 15     while (c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
 16     return w? -x: x;
 17 }
 18
 19 int const maxn=50003;
 20
 21 int B,n,a[maxn],t[300];
 22
 23 LL sum[300];
 24
 25 void update(int l,int r){
 26     int x=bl(l);
 27     int y=bl(r);
 28     if (x==y){
 29         if(t[x]>0) {
 30             F(i,l,r) if(a[i]>1) {
 31                 sum[x]-=a[i];
 32                 a[i]=(int)sqrt(a[i]);
 33                 sum[x]+=a[i];
 34                 if(a[i]==0 || a[i]==1) t[x]--;
 35             }
 36         }
 37     }else {
 38         if(t[x]>0){
 39             F(i,l,ed(x)) if(a[i]>1){
 40                 sum[x]-=a[i];
 41                 a[i]=(int) sqrt(a[i]);
 42                 sum[x]+=a[i];
 43                 if(a[i]==0 || a[i]==1) t[x]--;
 44             }
 45         }
 46         if(t[y]>0){
 47             F(i,st(y),r) if(a[i]>1){
 48                 sum[y]-=a[i];
 49                 a[i]=(int) sqrt(a[i]);
 50                 sum[y]+=a[i];
 51                 if(a[i]==0 || a[i]==1) t[y]--;
 52             }
 53         }
 54         F(i,x+1,y-1){
 55             if(t[i]==0) continue;
 56             F(j,st(i),ed(i)) {
 57                 if(a[j]>1){
 58                     sum[i]-=a[j];
 59                     a[j]=(int) sqrt(a[j]);
 60                     sum[i]+=a[j];
 61                     if(a[j]==0 || a[j]==1) t[i]--;
 62                 }
 63             }
 64         }
 65     }
 66 }
 67
 68 LL query(int l,int r){
 69     int x=bl(l);
 70     int y=bl(r);
 71     LL ans=0;
 72     if ( x==y){
 73         F(i,l,r) ans+=a[i];
 74     }else {
 75         F(i,l,ed(x)) ans+=a[i];
 76         F(i,st(y),r) ans+=a[i];
 77         F(i,x+1,y-1) ans+=sum[i];
 78     }
 79     return ans;
 80 }
 81 int main(){
 82     n=read(); B=(int) sqrt(n);
 83     F(i,1,n) a[i]=read();
 84     F(i,1,bl(n)) {
 85         t[i]=ed(i)-st(i)+1;
 86         F(j,st(i),ed(i)) {
 87             sum[i]+=a[j];
 88             if(a[j]==0 || a[j]==1) t[i]--;
 89         }
 90     }
 91     F(i,1,n){
 92         int x=read();
 93         int l=read();
 94         int r=read();
 95         int c=read();
 96         if(x==0) update(l,r);
 97         else printf("%lld\n",query(l,r));
 98     }
 99     return 0;
100 }

View Code

转载于:https://www.cnblogs.com/ZJXXCN/p/9704072.html