CSUST校赛G（线段树）

题目链接:校赛G.
思路:线段树每个结点储存m个数（0代表该数不存在，1代表存在），其他常规线段树写法，注意查询时的去重。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;

int n,m,q;
int a[200010];
int tr[200010*4][62];
int add[200010*4];
int b1[62],b2[62],f[62];

vector<int>v;
void prime()
{
    for(int i=2;i<m;i++)
    {
        for(int j=2;j<=i;j++)
        {
            if(i%j==0)
            {
                if(i==j)v.push_back(i);
                else break;
            }
        }
    }
}

void push_up(int node)
{
    for(int i=0;i<=m;i++)
    {
        if(tr[node*2][i]==1||tr[node*2+1][i]==1)tr[node][i]=1;
        else tr[node][i]=0;
    }
}

void push_down(int node,int l,int r)
{
    if(add[node])
    {
        int mid=(l+r)/2;
        memset(b1,0,sizeof b1);
        memset(b2,0,sizeof b2);
        for(int i=0;i<=m;i++)
        {
            if(tr[node*2][i])b1[(i+add[node])%m]=1;
            if(tr[node*2+1][i])b2[(i+add[node])%m]=1;
        }
        for(int i=0;i<=m;i++)
        {
            tr[node*2][i]=b1[i];
            tr[node*2+1][i]=b2[i];
        }
        add[node*2]=(add[node]%m+add[node*2]%m)%m;
        add[node*2+1]=(add[node]%m+add[node*2+1]%m)%m;
        add[node]=0;
    }
}

void build(int node,int l,int r)
{
    if(l==r)
    {
        tr[node][a[l]%m]=1;
        return ;
    }
    int mid=(l+r)/2;
    build(node*2,l,mid);
    build(node*2+1,mid+1,r);
    push_up(node);
}

void modify(int a,int b,int c,int node,int l,int r)
{
    if(a<=l&&b>=r)
    {
        memset(b1,0,sizeof b1);
        for(int i=0;i<=m;i++)
        {
            if(tr[node][i])b1[(i+c)%m]=1;
        }
        for(int i=0;i<=m;i++)tr[node][i]=b1[i];
        add[node]=(add[node]%m+c%m)%m;
    }
    else
    {
        push_down(node,l,r);
        int mid=(l+r)/2;
        if(a<=mid)modify(a,b,c,node*2,l,mid);
        if(b>=mid+1)modify(a,b,c,node*2+1,mid+1,r);
        push_up(node);
    }
}

void query(int a,int b,int node,int l,int r)
{
    if(a<=l&&b>=r)
    {
        int ans=0;
        for(int i=0;i<v.size();i++)
        {
            int x=v[i];
            if(tr[node][x])f[x]=1;
        }
        return ;
    }
    push_down(node,l,r);
    int mid=(l+r)/2;
    if(mid>=a)query(a,b,node*2,l,mid);
    if(mid+1<=b)query(a,b,node*2+1,mid+1,r);
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    prime();
    for(int i=1;i<=n;i++)cin>>a[i],a[i]=a[i]%m;
    build(1,1,n);
    //for(int i=1;i<=20;i++)cout<<tr[i][2]<<" ";
    cin>>q;
    while(q--)
    {
        int ch;cin>>ch;
        if(ch==1)
        {
            int l,r,v;cin>>l>>r>>v;
            modify(l,r,v%m,1,1,n);
        }
        else
        {
            memset(f,0,sizeof f);
            int l,r;cin>>l>>r;
            query(l,r,1,1,n);
            int ans=0;
            for(int i=0;i<=m;i++)ans+=f[i];
            cout<<ans<<"\n";
        }
    }
    return 0;
}

bitset优化,对fw来说这真的是个好神奇的操作…
神奇的| &的操作加上<< >>;

int n,m,q;
int add[200010*4];
bitset<60>tr[200010*4];
bitset<60>pr;

void prime()
{
    for(int i=2;i<m;i++)
    {
        for(int j=2;j<=i;j++)
        {
            if(i%j==0)
            {
                if(i==j)pr[i]=1;
                break;
            }
        }
    }
}

void push_up(int node)
{
    tr[node]=tr[node*2]|tr[node*2+1];
}

void lazy(int node,int v)
{
    tr[node]=tr[node]<<v | tr[node]>>(m-v);
}

void push_down(int node)
{
    if(add[node])
    {
        lazy(node*2,add[node]%m);
        lazy(node*2+1,add[node]%m);
        add[node*2]=(add[node]%m+add[node*2]%m)%m;
        add[node*2+1]=(add[node]%m+add[node*2+1]%m)%m;
        add[node]=0;
    }
}

void build(int node,int l,int r)
{
    if(l==r)
    {
        int x;cin>>x;
        tr[node][x%m]=1;
        return ;
    }
    int mid=(l+r)/2;
    build(node*2,l,mid);
    build(node*2+1,mid+1,r);
    push_up(node);
}

void modify(int a,int b,int c,int node,int l,int r)
{
    if(a<=l&&b>=r)
    {
        lazy(node,c%m);
        add[node]=(add[node]%m+c%m)%m;
    }
    else
    {
        push_down(node);
        int mid=(l+r)/2;
        if(a<=mid)modify(a,b,c,node*2,l,mid);
        if(b>=mid+1)modify(a,b,c,node*2+1,mid+1,r);
        push_up(node);
    }
}

bitset<60> query(int a,int b,int node,int l,int r)
{
    if(a<=l&&b>=r)return tr[node]&pr;
    push_down(node);
    bitset<60>s;
    int mid=(l+r)/2;
    if(a<=mid)s=s|query(a,b,node*2,l,mid);
    if(b>=mid+1)s=s|query(a,b,node*2+1,mid+1,r);
    return s&pr;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);

    cin>>n>>m;
    prime();
    build(1,1,n);

    cin>>q;
    while(q--)
    {
        int ch;cin>>ch;
        if(ch==1)
        {
            int l,r,v;cin>>l>>r>>v;
            modify(l,r,v,1,1,n);
        }
        else
        {
            int l,r;cin>>l>>r;
            cout<<query(l,r,1,1,n).count()<<"\n";
        }
    }
    return 0;
}