loj #6278. 数列分块入门 2

#6278. 数列分块入门 2

 

题目描述

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

输入格式

第一行输入一个数字 nnn。

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

接下来输入 nnn 行询问，每行输入四个数字 optmathrm{opt}opt、lll、rrr、ccc，以空格隔开。

若 opt=0mathrm{opt} = 0opt=0，表示将位于 [l,r][l, r][l,r] 的之间的数字都加 ccc。

若 opt=1mathrm{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操作

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 50010
using namespace std;
int a[maxn],b[maxn],add[maxn],pos[maxn];
int n,block,num;
int findl(int id){return (id-1)*block+1;}
int findr(int id){return min(id*block,n);}
void set(int id){
    int l=findl(id),r=findr(id);
    for(int i=l;i<=r;i++)a[i]+=add[id],b[i]=a[i];
    sort(b+l,b+r+1);
    add[id]=0;
}
void modify(int l,int r,int c){
    for(int i=l;i<=min(findr(pos[l]),r);i++)a[i]+=c;set(pos[l]);
    if(pos[l]==pos[r])return;
    if(pos[l]!=pos[r]){for(int i=findl(pos[r]);i<=r;i++)a[i]+=c;set(pos[r]);}
    int L=findr(pos[l])+1,R=findl(pos[r])-1;
    for(int i=pos[L];i<=pos[R];i++)add[i]+=c;
}
int find(int l,int r,int c){
    int pos=l-1,L=l;
    while(l<=r){
        int mid=(l+r)>>1;
        if(b[mid]<c)pos=mid,l=mid+1;
        else r=mid-1;
    }
    return pos-L+1;
}
int query(int l,int r,int c){
    int res=0;
    for(int i=l;i<=min(findr(pos[l]),r);i++)res+=(a[i]+add[pos[i]]<c);
    if(pos[l]==pos[r])return res;
    if(pos[l]!=pos[r]){for(int i=findl(pos[r]);i<=r;i++)res+=(a[i]+add[pos[i]]<c);}
    int L=findr(pos[l])+1,R=findl(pos[r])-1;
    for(int i=pos[L];i<=pos[R];i++)
        res+=find(findl(i),findr(i),c-add[i]);
    return res;
}
int main(){
    scanf("%d",&n);
    block=sqrt(n);
    for(int i=1;i<=n;i++)pos[i]=(i-1)/block+1;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    num=pos[n];for(int i=1;i<=num;i++)set(i);
    int op,l,r,c;
    for(int i=1;i<=n;i++){
        scanf("%d%d%d%d",&op,&l,&r,&c);
        if(op==0)modify(l,r,c);
        else printf("%d
",query(l,r,c*c));
    }
    return 0;
}