数列分块模板

数列分块入门1

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <string>
#include <set>
#include <cmath>
#include <map>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn = 1e5+7;
const int maxn2= 50005;
const int inf =0x3f3f3f3f;
int head[maxn],r[maxn],l[maxn],lazy[maxn];
int n,a[maxn],sk;
vector<int>q[maxn];
void build_k(){
    sk=sqrt(n);
    int num=ceil((double)n/sk);
    for(int i=1;i<=num;i++){
        l[i]=(i-1)*sk+1;r[i]=i*sk;lazy[i]=0;
    }r[num]=n;
    for(int i=1;i<=n;i++){
        head[i]=(i-1)/sk+1;
        q[head[i]].push_back(a[i]);
    }
}
void change(int nl,int nr,int pos){
    int f=head[nl],la=head[nr];
    int mid=min(r[f],nr);
    for(int i=nl;i<=mid;i++){
        a[i]+=pos;
    }q[f].clear();
    for(int i=l[f];i<=r[f];i++) q[f].push_back(a[i]);
    if(head[nl]==head[nr]) return ;
    for(int i=l[la];i<=nr;i++){
        a[i]+=pos;
    }q[la].clear();
    for(int i=l[la];i<=r[la];i++) q[la].push_back(a[i]);
    for(int i=f+1;i<la;i++) lazy[i]+=pos;
}
int query(int d){
    return a[d]+lazy[head[d]];
}
int main()
{
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        build_k();
        for(int i=1;i<=n;i++){
            int b,c,d,e;scanf("%d%d%d%d",&b,&c,&d,&e);
            if(b==0){
                change(c,d,e);
            }else{
                int ans=query(d);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}

分块2

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <string>
#include <set>
#include <cmath>
#include <map>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn = 1e5+7;
const int maxn2= 50005;
const int inf =0x3f3f3f3f;
int head[maxn],r[maxn],l[maxn],lazy[maxn];
int n,a[maxn],sk;
vector<int>q[maxn];
void build_k(){
    sk=sqrt(n);
    int num=ceil((double)n/sk);
    for(int i=1;i<=num;i++){
        l[i]=(i-1)*sk+1;r[i]=i*sk;lazy[i]=0;
    }r[num]=n;
    for(int i=1;i<=n;i++){
        head[i]=(i-1)/sk+1;
        q[head[i]].push_back(a[i]);
    }
    for(int i=1;i<=num;i++){
        sort(q[i].begin(),q[i].end());
    }
}
void change(int nl,int nr,int pos){
    int f=head[nl],la=head[nr];
    int mid=min(r[f],nr);
    for(int i=nl;i<=mid;i++){
        a[i]+=pos;
    }q[f].clear();
    for(int i=l[f];i<=r[f];i++) q[f].push_back(a[i]);
    sort(q[f].begin(),q[f].end());
    if(head[nl]==head[nr]) return ;
    for(int i=l[la];i<=nr;i++){
        a[i]+=pos;
    }q[la].clear();
    for(int i=l[la];i<=r[la];i++) q[la].push_back(a[i]);
    sort(q[la].begin(),q[la].end());
    for(int i=f+1;i<la;i++) lazy[i]+=pos;
}
int query(int nl,int nr,int pos){
    int ans=0;
    int f=head[nl],la=head[nr];
    int mid=min(r[f],nr);
    for(int i=nl;i<=mid;i++)  if(a[i]+lazy[f]<pos)  ans++;
    if(f==la) return ans;
    for(int i=l[la];i<=nr;i++) if(a[i]+lazy[la]<pos) ans++;
    for(int i=f+1;i<la;i++){
        ans+=lower_bound(q[i].begin(),q[i].end(),pos-lazy[i])-q[i].begin();
    }
    return ans;
}
int main()
{
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        build_k();
        for(int i=1;i<=n;i++){
            int b,c,d,e;scanf("%d%d%d%d",&b,&c,&d,&e);
            if(b==0){
                change(c,d,e);
            }else{
                int ans=query(c,d,e*e);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}