分块(n根n复杂度)

这是这几天才学的一个新的数据结构,因为时间问题,本人还没有进行实际操作过,所以这里只能给大家带来黄学长的大佬级代码,反正我一看就懂(蒟蒻还加了一些注释):

题目传送门: P2801 教主的魔法

大佬的代码:

//研究一发黄学长的代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int n,q,m,block;//n是题目中的,q也是,m是总块数,block是块的大小
int a[1000001],b[1000001],pos[1000001],add[1000001];//a是原始数组,b是分块中的数组,pos是每一个人分别在哪个块中记录,add是块的标记
void reset(int x)//分块初始化操作 ,修改不完整块也要用这个操作,方便后面的二分查找
{
    int l=(x-1)*block+1,r=min(x*block,n);//查找右边界 
    for(int i=l;i<=r;i++)
        b[i]=a[i];
    sort(b+l,b+r+1);
}
int find(int x,int v)//查询操作,二分
{
    int l=(x-1)*block+1,r=min(x*block,n);
    int last=r;
    while(l<=r)//二分一下,完美结束! 
    {
        int mid=(l+r)/2;
        if(b[mid]<v)l=mid+1;
        else r=mid-1;
    }
    return last-l+1;
}
void update(int x,int y,int v)//修改操作
{
    if(pos[x]==pos[y])//如果修改都在同一个块中 ,暴力修改 
    {
        for(int i=x;i<=y;i++)a[i]=a[i]+v;
    }
    else 
    {
        for(int i=x;i<=pos[x]*block;i++)a[i]=a[i]+v;//修改x及所在块的后面, 
        for(int i=(pos[y]-1)*block+1;i<=y;i++)a[i]=a[i]+v;//修改y及其所在块的前面 
    }
    reset(pos[x]);reset(pos[y]);
    for(int i=pos[x]+1;i<pos[y];i++)
       add[i]+=v;
}
int query(int x,int y,int v)//总的查询操作
{
    int sum=0;
    if(pos[x]==pos[y])//如果在同一个块中,暴力!!! 
    {
        for(int i=x;i<=y;i++)if(a[i]+add[pos[i]]>=v)sum++;
    }
    else 
    {
        for(int i=x;i<=pos[x]*block;i++)//处理x所在的残块,暴力!!! 
            if(a[i]+add[pos[i]]>=v)sum++;
        for(int i=(pos[y]-1)*block+1;i<=y;i++)//处理y所在的残块,暴力!!! 
            if(a[i]+add[pos[i]]>=v)sum++;
    }
    for(int i=pos[x]+1;i<pos[y];i++)//处理x块和y块中间的完整的块。 
        sum+=find(i,v-add[i]);
    return sum;
}
int main()
{
    scanf("%d%d",&n,&q);
    block=int(sqrt(n));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        pos[i]=(i-1)/block+1;//这句话的意思是就是查找第i个英雄在第几个块 
        b[i]=a[i];
    }
    if(n%block)m=n/block+1;//有几个块,不为零说明存在不完整的块,所以说要+1 
    else m=n/block;
    for(int i=1;i<=m;i++)reset(i);//初始化块 
    for(int i=1;i<=q;i++)
    {
        char ch[5];int x,y,v;
        scanf("%s%d%d%d",ch,&x,&y,&v);
        if(ch[0]=='M')update(x,y,v);
        else printf("%d\n",query(x,y,v));
    }
    return 0;
}

注意:

经过本人亲自打了一遍,我发现易错的地方有如下两点:

1、查询残块的时候一定不要把左右区间弄反了。

2、就是在分配第几个人在第几个块的时候一定要把编号-1!!!!

posted @ 2020-07-17 13:12  Mudrobot  阅读(182)  评论(0编辑  收藏  举报