线段树练习4加强版 CODEVS - 5037

http://codevs.cn/problem/5037/

分块法 其实就是分块进行暴力 n*sqrt(n)的复杂度

这题就他妈的SB 一直T 把结构体去了就过了 我不就用了个成员调用吗 至于这样搞我吗 真TM的跪了

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;

int val[510][200010];
int ary[200010],mp[200010],laz[510];
int n,q,k,len;

void update(int l,int r,int v)
{
    int i;
    for(i=l;i<=r&&mp[i]==mp[l];i++)
    {
        val[mp[l]][ary[i]]--;
        val[mp[l]][(ary[i]+v)%k]++;
        ary[i]=(ary[i]+v)%k;
    }
    for(i=mp[l]+1;i<=mp[r]-1;i++)
    {
        laz[i]=(laz[i]+v)%k;
    }
    for(i=r;i>=l&&mp[i]!=mp[l]&&mp[i]==mp[r];i--)
    {
        val[mp[r]][ary[i]]--;
        val[mp[r]][(ary[i]+v)%k]++;
        ary[i]=(ary[i]+v)%k;
    }
}

int query(int l,int r)
{
    int res,i;
    res=0;
    for(i=l;i<=r&&mp[i]==mp[l];i++)
    {
        if((ary[i]+laz[mp[l]])%k==0) res++;
    }
    for(i=mp[l]+1;i<=mp[r]-1;i++)
    {
        res+=val[i][(k-laz[i])%k];
    }
    for(i=r;i>=l&&mp[i]!=mp[l]&&mp[i]==mp[r];i--)
    {
        if((ary[i]+laz[mp[r]])%k==0) res++;
    }
    return res;
}

int main()
{
    int i,j,l,r,v;
    char op[10];
    scanf("%d%d%d",&n,&q,&k);
    len=sqrt(n);
    for(i=0;i<n;i++)
    {
        scanf("%d",&ary[i]);
        ary[i]%=k;
        mp[i]=i/len;
        val[mp[i]][ary[i]]++;
    }
    while(q--)
    {
        scanf("%s",op);
        if(op[0]=='a')
        {
            scanf("%d%d%d",&l,&r,&v);
            l--,r--,v%=k;
            update(l,r,v);
        }
        else
        {
            scanf("%d%d",&l,&r);
            l--,r--;
            printf("%d\n",query(l,r));
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sunyutian1998/article/details/81416026