题意为给定一个数字n,每次当n>1时,将n分拆为n/2,n%2,n/2。再给定区间l,r求分拆后l,r内1的个数。
由题意可看出,这道题明显时需要递归求解的。不过递归的顺序决定了计算答案的快慢。
第一种方法就是,不断从左向右递归求解,一个一个进行编号。至少复杂度为o(n),明显不可能实现。
这时我们观察到n只有10的五次方大小,便可以需要先计算出当前数字所更新的区间范围,如果已经不属于待求区间便可以直接返回。这样有极大的时间以及空间浪费省去了。复杂度为logn*10的五次方,轻松AC。
#include<iostream>
using namespace std;
long long n,l,r;
long long L=1,R=1;
long long ans=0;
void dfs(long long now,long long le,long long ri)
{
if(le>r || ri<l || now==0)
return;
if(now==1)
{
ans++;
return;
}
long long mid=(le+ri)/2;
dfs(now/2,le,mid-1);
dfs(now%2,mid,mid);
dfs(now/2,mid+1,ri);
}
int main()
{
cin>>n>>l>>r;
long long temp=n;
while(temp>1)
{
R=R*2+1;
temp/=2;
}
dfs(n,L,R);
cout<<ans<<endl;
}