题意:
已知对于任意的 X ,将其本身填入集合时,也需要将 [ x - lowbit(x) + 1 , x - 1] 重新填入集合。
给出 X 的最大值 N 和询问次数 K
共有两类询问
1.询问将 [ L , R ] 中所有数逐次填入集合时,一共对每个数字填入了多少次。
2.询问将 [ 1 , N ] 中所有数逐次填入集合时,数字 X 被填入了多少次。
思路:
注意到对于任意的 X 填入集合时,一共操作了 [ x - lowbit(x) + 1 , x - 1]的大小 + 1 个数,即 lowbit(x) 个数。
由此我们可以做出图,形似树状数组。
那么对于操作1
如果我们可以计算出lowbit(i)的前 N 项和,我们就可以通过前缀和的思想O(1)来求出答案。
如何求出lowbit的前N项和呢?
我们注意到对于任意一个数 X ,lowbit(X) = 以 X 为根的子树的子叶节点数。
我们又知道高度为 H 的满二叉树有 2^H 个子叶节点(H = 0,1,2.......)。
那么我们可以通过统计 1.......X 以内有多少个高度为 0,1,2.....的子树就可以计算出lowbit的前 X 项和是多少。统计的过程显然是 O(logX) 的。
所以对于操作1,其复杂度可以控制在O(logN)左右。
对于操作2,由图显然对于任意一个数 X 其插入的次数是其父节点的个数
例 1 4、6 2 、8 1
利用 lowbit 递加即可 复杂度为 O(logN)
以上本题的复杂度为 O(KlogN)
代码:
#include <bits/stdc++.h>
using namespace std;
long long lowbit(long long x){
return x&(-x);
}
long long solve(long long x){
long long ans=0,two=1;
while(x!=0){
ans+=two*(x-x/2);
x/=2;
two*=2;
}
return ans;
}
int main()
{
long long n,x,y,ans;
int m,ch;
while(scanf("%lld%d",&n,&m)!=-1){
for(int i=0;i<m;i++){
scanf("%d",&ch);
if(ch==1){
scanf("%lld%lld",&x,&y);
ans=solve(y)-solve(x-1);
}else{
scanf("%lld",&x);
ans=0;
while(x<=n){
ans++;
x+=lowbit(x);
}
}
printf("%lld\n",ans);
}
}
}