[BZOJ 2568]比特集合

这道题的思路还是比较好想的喵~

首先令数组 C[k][num] 表示 2 进制最后 k 位 <=num 的数的个数

查询第 k 位为 1 即询问 C[k][(1<<k+1)-1]-C[k][(1<<k)-1] 的值

因为要支持插入、删除，C 数组应使用树状数组或线段树来维护

然后可以用类似于标记永久化的方法，把 ADD 操作搞掉喵~

但是细节部分处理起来就比较麻烦了，WA 了好多次莫名其妙的 A 了

我也不敢写太多……

#include <cstdio>
#include <map>

namespace IOspace
{
    inline char getch()
    {
        register char ch;
        do ch=getchar(); while (ch!='A' && ch!='I' && ch!='D' && ch!='Q');
        return ch;
    }
    inline int getint()
    {
        register int num=0;
        register char ch;
        do ch=getchar(); while (ch<'0' || ch>'9');
        do num=num*10+ch-'0', ch=getchar(); while (ch>='0' && ch<='9');
        return num;
    }
    inline void putint(int num, char ch='
')
    {
        char stack[15];
        register int top=0;
        if (num==0) stack[top=1]='0';
        for ( ;num;num/=10) stack[++top]=num%10+'0';
        for ( ;top;top--) putchar(stack[top]);
        if (ch) putchar(ch);
    }
}

int N;
int add;
std::map<int, int> s;
int sum[17][65537];
inline int lim(int x) {return (1<<x+1)-1;}
inline int min(int x, int y) {return x<y?x:y;}
inline int max(int x, int y) {return x>y?x:y;}
inline int lowbit(int x) {return x & -x;}
inline void update(int, int, int);
inline int query(int, int);

int main()
{
    N=IOspace::getint();
    for ( ;N;N--)
    {
        char ch=IOspace::getch(); int x=IOspace::getint();
        if (ch=='A') add+=x;
        if (ch=='I')
        {
            x-=add;
            s[x]++;
            for (int i=0;i<16;i++) update(i, (x & lim(i))+1, 1);
        }
        if (ch=='D')
        {
            x-=add;
            int t=s[x];
            s[x]=0;
            for (int i=0;i<16;i++) update(i, (x & lim(i))+1, -t);
        }
        if (ch=='Q')
        {
            int l=1<<x, r=lim(x);
            int ans=0;
            ans+=query(x, min(max(r-(add & lim(x))+1, 0), 1<<16));
            ans-=query(x, min(max(l-(add & lim(x)), 0), 1<<16));
            l+=1<<x+1, r+=1<<x+1;
            ans+=query(x, min(max(r-(add & lim(x))+1, 0), 1<<16));
            ans-=query(x, min(max(l-(add & lim(x)), 0), 1<<16));
            IOspace::putint(ans);
        }
    }

    return 0;
}
inline void update(int k, int i, int v)
{
    for ( ;i<=1<<16;i+=lowbit(i)) sum[k][i]+=v;
}
inline int query(int k, int i)
{
    int ret=0;
    for ( ;i;i-=lowbit(i)) ret+=sum[k][i];
    return ret;
}