题目链接:http://www.lydsy.com:808/JudgeOnline/problem.php?id=2568
题意:维护一个集合S,支持以下操作:
(1)INS M : 将元素 M 插入到集合S中;
(2)DEL M : 将集合S中所有等于 M 的元素删除;(3)ADD M : 将集合S中的所有元素都增加数值M ;(4)QBIT k : 查询集合中有多少个元素满足其二进制的第 k位为 1 。
思路:
(1) ADD 操作的那个和单独拿出来,设为sum,集合S中的每个元素x实际值为x+sum;
(2)设f[k][t]表示第k位为1,且这个数字小于等于t的数字的个数。每次查询时,设L=2^x,R=2^(x+1)-1,则答案为f[k][R]-f[k][L-1];
(3)由于增加一个数字时这个f值是成段改变,因此要用树状数组维护这个f数组;
(4)对于那些插入的数字都是多少以及每个数字有多少个,用一个map记录,这样删除时就知道在树状数组要减去多少。
我当时有个问题没有明白,因为插入x时实际要插入的数字是x-sum,那么x-sum为负数时这个位跟正数的位不太一样。负数的二进制表示是对应正数的二进制表示取反加1。
比如
-1=11111111 11111111 11111111 11111111
-2=11111111 11111111 11111111 11111110
-3=11111111 11111111 11111111 11111101
-4=11111111 11111111 11111111 11111100
直接插入(下面可以看到,不需要特殊考虑负数)
后面那还加了个1是因为树状数组里下标都是从1开始的。
然后求和时是这样的
这个分为两部分,第一部分:计算的是[L,R]区间,设k=2,那么二进制表示L=100,R=111。设sum=1011,那么实际要计算的区间为[001,100],只要一个数字的后三位在这个区间,即[001,100],那么它加上sum之后的后三位都会落到[L,R]区间。其实这个是没有进位的。
我们再设sum=1110,其他不变,那么上面的实际求和区间变成[000,001]。我们发现,除了这个区间,[110,111]这个区间也是可以的。这个其实是进位产生的,进位之后求和区间由[100,111]变为[1100,1111],这样减去sum的后三位110实际区间为[110,1001],我们发现1001,1000都不会有这个值,所以实际就是[110,111]。这就是上面求和的第二部分。
那么一个负数加上sum之后也可能到达这个区间,sum=1110,[-10,-7],这些负数的二进制为
-10=11111111 11111111 11111111 11110110
-9 =11111111 11111111 11111111 11110111
-8 =11111111 11111111 11111111 11111000
-7 =11111111 11111111 11111111 11111001
我们发现,后三位都在计算的两个区间里。所以负数不需要额外考虑。
int S[20][N];mapmp; int n; void add(int k,int x,int t){ while(x