Digits Count FZU - 2105

点击打开链接

数据量很小 [0,15] 四个二进制位 每一位对应一颗线段树 然后就是对每一位进行区间更新与查询

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 0x3f3f3f3f

struct node
{
    int l;
    int r;
    short int laz1;
    short int laz2;
    int val;
};

node tree[4][4000010];
int n;

void pushup(int id,int cur)
{
    tree[id][cur].val=tree[id][2*cur].val+tree[id][2*cur+1].val;
    return;
}

void pushdown(int id,int cur)
{
    int t;
    t=tree[id][cur].laz1;
    if(t!=-1)
    {
        tree[id][2*cur].val=(tree[id][2*cur].r-tree[id][2*cur].l+1)*t;
        tree[id][2*cur].laz1=tree[id][cur].laz1;
        tree[id][2*cur].laz2=0;
        tree[id][2*cur+1].val=(tree[id][2*cur+1].r-tree[id][2*cur+1].l+1)*t;
        tree[id][2*cur+1].laz1=tree[id][cur].laz1;
        tree[id][2*cur+1].laz2=0;
        tree[id][cur].laz1=-1;
    }
    if(tree[id][cur].laz2!=0)
    {
        tree[id][2*cur].val=(tree[id][2*cur].r-tree[id][2*cur].l+1-tree[id][2*cur].val);
        tree[id][2*cur].laz2^=1;
        tree[id][2*cur+1].val=(tree[id][2*cur+1].r-tree[id][2*cur+1].l+1-tree[id][2*cur+1].val);
        tree[id][2*cur+1].laz2^=1;
        tree[id][cur].laz2=0;
    }
    return;
}

void build(int l,int r,int cur)
{
    int m,val,i;
    for(i=0;i<4;i++)
    {
        tree[i][cur].l=l;
        tree[i][cur].r=r;
        tree[i][cur].laz1=-1;
        tree[i][cur].laz2=0;
    }
    if(l==r)
    {
        scanf("%d",&val);
        for(i=0;i<4;i++)
        {
            tree[i][cur].val=val%2;
            val/=2;
        }
        return;
    }
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
    for(i=0;i<4;i++) pushup(i,cur);
    return;
}

void updateI(int pl,int pr,int val,int id,int cur)
{
    if(pl<=tree[id][cur].l&&tree[id][cur].r<=pr)
    {
        tree[id][cur].val=(tree[id][cur].r-tree[id][cur].l+1)*val;
        tree[id][cur].laz1=val;
        tree[id][cur].laz2=0;
        return;
    }
    pushdown(id,cur);
    if(pl<=tree[id][2*cur].r) updateI(pl,pr,val,id,2*cur);
    if(pr>=tree[id][2*cur+1].l) updateI(pl,pr,val,id,2*cur+1);
    pushup(id,cur);
    return;
}

void updateII(int pl,int pr,int id,int cur)
{
    if(pl<=tree[id][cur].l&&tree[id][cur].r<=pr)
    {
        tree[id][cur].val=(tree[id][cur].r-tree[id][cur].l+1-tree[id][cur].val);
        tree[id][cur].laz2^=1;
        return;
    }
    pushdown(id,cur);
    if(pl<=tree[id][2*cur].r) updateII(pl,pr,id,2*cur);
    if(pr>=tree[id][2*cur+1].l) updateII(pl,pr,id,2*cur+1);
    pushup(id,cur);
    return;
}

int query(int pl,int pr,int id,int cur)
{
    int res;
    if(pl<=tree[id][cur].l&&tree[id][cur].r<=pr)
    {
        return tree[id][cur].val;
    }
    pushdown(id,cur);
    res=0;
    if(pl<=tree[id][2*cur].r) res+=query(pl,pr,id,2*cur);
    if(pr>=tree[id][2*cur+1].l) res+=query(pl,pr,id,2*cur+1);
    return res;
}

int main()
{
    int t,q,i,val,l,r,res;
    char op[10];
    scanf("%d",&t);
    while(t--)
    {
        memset(tree,0,sizeof(tree));
        scanf("%d%d",&n,&q);
        build(1,n,1);
        while(q--)
        {
            scanf("%s",op);
            if(op[0]=='A')//&0
            {
                scanf("%d%d%d",&val,&l,&r);
                l++,r++;
                for(i=0;i<4;i++)
                {
                    if(!((1<<i)&val)) updateI(l,r,0,i,1);
                }
            }
            else if(op[0]=='O')//|1
            {
                scanf("%d%d%d",&val,&l,&r);
                l++,r++;
                for(i=0;i<4;i++)
                {
                    if((1<<i)&val) updateI(l,r,1,i,1);
                }
            }
            else if(op[0]=='X')
            {
                scanf("%d%d%d",&val,&l,&r);
                l++,r++;
                for(i=0;i<4;i++)
                {
                    if((1<<i)&val) updateII(l,r,i,1);
                }
            }
            else
            {
                scanf("%d%d",&l,&r);
                l++,r++;
                res=0;
                for(i=0;i<4;i++)
                {
                    res+=(1<<i)*query(l,r,i,1);
                }
                printf("%d\n",res);
            }
        }
    }
    return 0;
}