Amount of Degrees

https://loj.ac/problem/10163

题目描述

  求一段区间\([x,y]\)内满足恰好等于\(K\)个不相等的\(B\)的整次幂的和的数的个数。

思路

  题目求的其实就是在区间\([x,y]\)内满足在\(B\)进制下有且仅有\(K\)个\(1\)，其余位位\(0\)的数的个数。我们显然可以先转化为区间减法，接下来考虑如何求\([1,x]\)内符合条件的数的个数。我们可以按位思考，如果这一位大于\(1\)，显然对于这之后的位数可以任意选，我们直接统计剩余的位数中选出\(k\)个\(1\)的方案数即可，如果这一位等于\(1\)，我们可以考虑先加上这一位为\(0\)时的方案数，再令这一位为\(1\)统计答案即可。

代码

#include<bits/stdc++.h>
using namespace std;

int f[40][40];
void pre()
{
    f[0][0]=1;
    for(int i=1;i<=31;i++)
    {
        f[i][0]=f[i-1][0];
        for(int j=1;j<=i;j++)
            f[i][j]=f[i-1][j]+f[i-1][j-1];
    }
}
int p[50],b;
int cal(int x,int k)
{
    int cnt=0;
    while(x)p[++cnt]=x%b,x/=b;
    int ans=0;
    for(int i=cnt;i>=1;i--)
    {
        if(p[i]>1){ans+=f[i][k];break ;}
        else if(p[i]==1)ans+=f[i-1][k],k--;
        if(k<0)break ;
    }
    return ans;
}

int main()
{
    int x,y,k;
    pre();
    scanf("%d%d%d%d",&x,&y,&k,&b);
    printf("%d",cal(y+1,k)-cal(x,k));
}