题意:给出四个数,x,y,k,b,求出区间[x,y]内能表示成k个不相等的b次幂之和的数的数量。例: x=15,y=20,k=2,b=2
17=2^4+2^0
18=2^4+2^1
20=2^4+2^2
总共有3个数满足条件。
首先,所求解count(x,y)=count(0,y)-count(0,x);其次,容易想到,k个不相等的b次幂数相加,结果用b进制表示的话,必然是含有k个1,因此题目所求可以转换成求对应区间内的数转换为b进制后,含k个1的数的数量。这里首先考虑二进制的情况,即b=2.如何求区间[0,n]内数的二进制表示中,含k个1的数的数量呢?
用f[i][j]表示j位二进制位的数恰含j个1的数量,有f[i][j]=f[i-1][j]+f[i-1][j-1].可以用下图帮助理解,f[i][j]代表高度为i的子树中含j个1的数的数量.统计count(0,n)的时,从根节点出发,每遇到一个1,就统计上层结点左子树的结果。
树的最底层是第0层,子树的根节点不记入统计过程中,可以看出每个子树的统计结果都是一样的。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <cmath>
using namespace std;
const int N=1000;
int f[N][N],l,r,k,b;
void init(void) // init
{
int i,j;
f[0][0]=1;
for (i=1;i<32;i++) {
f[i][0]=f[i-1][0];
for (j=1;j<=i;j++) {
f[i][j]=f[i-1][j]+f[i-1][j-1];
}
}
}
int slove(int n)
{
int t=n,tot=0,cnt=0;
int s[N],top=0,i;
while(t){
s[top++]=t%b;
t/=b;
}
int ans=0;
for (i=top-1;i>=0;i--) {
if (s[i]>1) {
ans+=f[i+1][k-tot];
break;
}
if (s[i]==1) {
ans+=f[i][k-tot];
tot++;
if (tot==k) break;
}
}
if (tot==k) ans++; // 判断n本身
return ans;
}
int main()
{
init();
//freopen("in.txt","r",stdin);
while(cin>>l>>r>>k>>b)
{
cout<<slove(r)-slove(l-1)<<endl;
}
return 0;
}