分块试水--CODEVS5037 线段树练习4加强版

感觉这才算入门题吧。。前面那些线段树练习，改几个字符就过了一定要搞成几道题。。

n<=2e5的数列，给常数K<=2e5，m<=2e5个操作，区间加，问一个区间里K的倍数。

这题空间？？？当成512MB吧。。n*sqrt(n)开得下

每个块记cnt(x,i)表示第x个块有多少数%K=i，以及整块+标记add(x)，然后回答询问时默认一个块里的数都加上了add(x)即可，修改常规。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<stdlib.h>
#include<math.h>
//#include<iostream>
using namespace std;

int n,m,q,K;
#define maxn 200011
#define maxm 461
int cnt[maxm][maxn],a[maxn],bel[maxn],tot,add[maxm];
void modify(int x,int y,int v)
{
    if (bel[x]==bel[y])
        for (int i=x;i<=y;i++) cnt[bel[i]][a[i]]--,a[i]+=v,a[i]%=K,cnt[bel[i]][a[i]]++;
    else
    {
        for (int i=bel[x]+1;i<bel[y];i++) add[i]+=v,add[i]%=K;
        for (int i=x;bel[i]==bel[x];i++) cnt[bel[i]][a[i]]--,a[i]+=v,a[i]%=K,cnt[bel[i]][a[i]]++;
        for (int i=y;bel[i]==bel[y];i--) cnt[bel[i]][a[i]]--,a[i]+=v,a[i]%=K,cnt[bel[i]][a[i]]++;
    }
}
int query(int x,int y)
{
    int ans=0;
    if (bel[x]==bel[y])
        for (int i=x;i<=y;i++) ans+=(a[i]+add[bel[i]]==0 || a[i]+add[bel[i]]==K);
    else
    {
        for (int i=bel[x]+1;i<bel[y];i++) ans+=cnt[i][add[i]?K-add[i]:0];
        for (int i=x;bel[x]==bel[i];i++) ans+=(a[i]+add[bel[i]]==0 || a[i]+add[bel[i]]==K);
        for (int i=y;bel[y]==bel[i];i--) ans+=(a[i]+add[bel[i]]==0 || a[i]+add[bel[i]]==K);
    }
    return ans;
}

bool isdigit(char c) {return c>='0' && c<='9';}
int qread()
{
    char c;int s=0;while (!isdigit(c=getchar()));
    do s=s*10+c-'0'; while (isdigit(c=getchar())); return s;
}
int main()
{
    n=qread(),q=qread(),K=qread();
    m=(int)sqrt(n);
    for (int i=1;i<=n;i++) bel[i]=(i-1)/m+1;
    tot=bel[n];
    
    for (int i=1;i<=n;i++)
    {
        a[i]=qread()%K;
        cnt[bel[i]][a[i]]++;
    }
    
    char c;
    for (int i=1,x,y,z;i<=q;i++)
    {
        while (!((c=getchar())>='a' && c<='z'));
        if (c=='a')
        {
            x=qread(),y=qread(),z=qread();
            modify(x,y,z);
        }
        else
        {
            x=qread(),y=qread();
            printf("%d
",query(x,y));
        }
        while (c>='a' && c<='z') c=getchar();
    }
    return 0;
}