LOJ #6278. 数列分块入门 2

hzwer tql!

渐渐找到了分块的套路。

给出一个长为  n 的数列，以及 n 个操作，操作涉及区间加法，询问区间内小于某个值 c*c 的元素个数。

残块先在原数组上暴力，然后拿原数组更新块数组（效率高于结构体存id！），整块就标记。

code：

 1 #include<iostream>
 2 #include<cmath>
 3 #include<cstdio>
 4 #include<algorithm>
 5 #include<vector>
 6 #define pb push_back
 7 #define lb lower_bound
 8 using namespace std;
 9 
10 const int Maxn = 50005; 
11 
12 vector<int> V[Maxn];
13 int bl[Maxn],v[Maxn],atag[Maxn];
14 int n,blo,l,r,opt,c;
15 
16 void reset(int x){
17     for(int i = (x-1)*blo+1;i <= min(n,x*blo);i++)
18         V[x][(i-1)%blo] = v[i];
19     sort(V[x].begin(),V[x].end());
20 } 
21 
22 void add(int l,int r,int c){
23     if(bl[l] == bl[r]){
24         for(int i = l;i <= r;i++)v[i] += c;
25         reset(bl[r]);return;
26     }
27     for(int i = l;i <= min(n,bl[l]*blo);i++)v[i] += c;reset(bl[l]);
28     for(int i = (bl[r]-1)*blo+1;i <= r;i++)v[i] += c;reset(bl[r]);
29     for(int i = bl[l]+1;i < bl[r];i++)atag[i] += c;
30 }
31 
32 int ask(int l,int r,int c2){
33     if(l > r)return 0;
34     int cnt = 0;
35     if(bl[l] == bl[r]){
36         for(int i = l;i <= r;i++)if(v[i]+atag[bl[i]] < c2)cnt++;
37         return cnt;
38     }
39     for(int i = l;i <= min(bl[l]*blo,n);i++)
40         if(v[i]+atag[bl[i]] < c2)cnt++;
41     for(int i = (bl[r]-1)*blo+1;i <= min(r,n);i++)
42         if(v[i]+atag[bl[i]] < c2)cnt++;
43     for(int i = bl[l]+1;i < bl[r];i++)
44         cnt += lb(V[i].begin(),V[i].end(),c2-atag[i])-V[i].begin();
45     return cnt;
46 }
47 
48 int main(){
49     scanf("%d",&n);blo = sqrt(n);
50     for(int i = 1;i <= n;i++){
51         scanf("%d",&v[i]);
52         bl[i] = (i-1)/blo+1;
53         V[bl[i]].pb(v[i]);
54     }
55     for(int i = 1;i <= bl[n];i++)sort(V[i].begin(),V[i].end());
56     for(int i = 1;i <= n;i++){
57         scanf("%d%d%d%d",&opt,&l,&r,&c);
58         if(opt)printf("%d\n",ask(l,r,c*c));
59         else add(l,r,c);
60     }
61 return 0;
62 }