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] 的所有数字的和。

输出格式

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

样例

样例输入

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,−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

 

posted @ 2018-09-25 22:44  zjxxcn  阅读(253)  评论(0)  编辑  收藏  举报