loj #6278. 數列分塊入門 2

#6278. 數列分塊入門 2

 

題目描述

給出一個長爲 nnn 的數列，以及 nnn 個操做，操做涉及區間加法，詢問區間內小於某個值 xxx 的元素個數。

輸入格式

第一行輸入一個數字 nnn。

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

接下來輸入 nnn 行詢問，每行輸入四個數字 opt\mathrm{opt}opt、lll、rrr、ccc，以空格隔開。

若 opt=0\mathrm{opt} = 0opt=0，表示將位於 [l,r][l, r][l,r] 的之間的數字都加 ccc。

若 opt=1\mathrm{opt} = 1opt=1，表示詢問 [l,r][l, r][l,r] 中，小於 c2c^2c​2​​ 的數字的個數。

輸出格式

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

樣例

樣例輸入

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

樣例輸出

3
0
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。

改了半天終於改對了

第一處錯誤：

20和37行

第二處錯誤：

19和36行的取min操做

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<cmath>
 6 #define maxn 50010
 7 using namespace std;
 8 int a[maxn],b[maxn],add[maxn],pos[maxn];
 9 int n,block,num;
10 int findl(int id){return (id-1)*block+1;}
11 int findr(int id){return min(id*block,n);}
12 void set(int id){
13     int l=findl(id),r=findr(id);
14     for(int i=l;i<=r;i++)a[i]+=add[id],b[i]=a[i];
15     sort(b+l,b+r+1);
16     add[id]=0;
17 }
18 void modify(int l,int r,int c){
19     for(int i=l;i<=min(findr(pos[l]),r);i++)a[i]+=c;set(pos[l]);
20     if(pos[l]==pos[r])return;
21     if(pos[l]!=pos[r]){for(int i=findl(pos[r]);i<=r;i++)a[i]+=c;set(pos[r]);}
22     int L=findr(pos[l])+1,R=findl(pos[r])-1;
23     for(int i=pos[L];i<=pos[R];i++)add[i]+=c;
24 }
25 int find(int l,int r,int c){
26     int pos=l-1,L=l;
27     while(l<=r){
28         int mid=(l+r)>>1;
29         if(b[mid]<c)pos=mid,l=mid+1;
30         else r=mid-1;
31     }
32     return pos-L+1;
33 }
34 int query(int l,int r,int c){
35     int res=0;
36     for(int i=l;i<=min(findr(pos[l]),r);i++)res+=(a[i]+add[pos[i]]<c);
37     if(pos[l]==pos[r])return res;
38     if(pos[l]!=pos[r]){for(int i=findl(pos[r]);i<=r;i++)res+=(a[i]+add[pos[i]]<c);}
39     int L=findr(pos[l])+1,R=findl(pos[r])-1;
40     for(int i=pos[L];i<=pos[R];i++)
41         res+=find(findl(i),findr(i),c-add[i]);
42     return res;
43 }
44 int main(){
45     scanf("%d",&n);
46     block=sqrt(n);
47     for(int i=1;i<=n;i++)pos[i]=(i-1)/block+1;
48     for(int i=1;i<=n;i++)scanf("%d",&a[i]);
49     num=pos[n];for(int i=1;i<=num;i++)set(i);
50     int op,l,r,c;
51     for(int i=1;i<=n;i++){
52         scanf("%d%d%d%d",&op,&l,&r,&c);
53         if(op==0)modify(l,r,c);
54         else printf("%d\n",query(l,r,c*c));
55     }
56     return 0;
57 }