分块入门4: 区间修改+区间查询和

题目链接:
题目大意:
分块入门4: 区间修改+区间查询和_第1张图片
分块入门4: 区间修改+区间查询和_第2张图片
直接上代码:

#include
using namespace std;
typedef long long LL;
//由块号寻找第一个块元素的下标
#define LL(x) ((x-1)*Len+1)
const LL maxn=5e5+5;
LL read()
{
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

LL a[50050], b[50050], add[505], s[505];//b[i]:i的块号
LL Len, n;

void Add(LL L,LL R,LL x)
{
    LL bL=b[L], bR=b[R];

    if(bL==bR)
    {
        for(LL i=L;i<=R;i++)
        {
            a[i]+=x;
            s[bL]+=x;
        }
    }
    else
    {
        for(LL i=L;i<LL(bL+1);i++)
        {
            a[i]+=x;
            s[bL]+=x;
        }

        for(LL i=bL+1;i<bR;i++)
        {
            add[i]+=x;
            s[i]+=x*Len;
        }

        for(LL i=LL(bR);i<=R;i++)
        {
            a[i]+=x;
            s[bR]+=x;
        }
    }

}

LL query(LL L, LL R, LL c)
{
    LL ans=0;
    LL bL=b[L], bR=b[R];

    if(bL==bR)
    {
        for(LL i=L;i<=R;i++)
        {
            ans=(ans+a[i]+add[bL])%(c+1);
        }
    }
    else
    {
        for(LL i=L;i<LL(bL+1);i++)
        {
            ans=(ans+a[i]+add[bL])%(c+1);
        }

        for(LL i=bL+1;i<bR;i++)
        {
            ans=(ans+s[i])%(c+1);
        }

        for(LL i=LL(bR);i<=R;i++)
        {
            ans=(ans+a[i]+add[bR])%(c+1);
        }
    }

    return ans;
}

void build(LL n)
{
    Len=n/sqrt(n);
    for(LL i=1;i<=n;i++)
    {
        a[i]=read();
        add[i]=0;
        s[(i-1)/Len+1]=0;
    }
    for(LL i=1;i<=n;i++)
    {
        b[i]=(i-1)/Len+1;
        s[b[i]]+=a[i];
    }
}

int main()
{
    n=read();
    build(n);

    for(LL i=1;i<=n;i++)
    {
        LL op=read(), L=read(), R=read(), c=read();
        if(op==0)
        {
            Add(L, R, c);
        }
        else
        {
            printf("%lld\n", query(L, R, c));
        }
    }

    return 0;
}

你可能感兴趣的:(分块)