Amount of Degrees URAL - 1057 数位dp

题意：求（x--y）区间转化为b进制以后， 1 的个数为 k 的数的出现次数，并且其余位上的数必须为0。

原题要求的是k个不同b次幂相加得到的不同的数的个数，所以每一位上的的数不能出现除1和0以外的数，有了这样一个转换就好办了。

数位dp，枚举时只枚举含0,1的位，一定是从高位枚举，设d[pos][num1][sum][dig]为从pos到最低位，1出现次数为num1时，要求出现sum次，dig进制下数的个数。

注意前导0的判断，几乎是模板题，写法都差不多。

#include <iostream>  
#include <cstdio>  
#include <ctime>  
#include <cstring>  
#include <algorithm>  
#include <vector>  
#include <map>  
#include <cmath>  
#include <set>  
#include <queue>  
using namespace std;  
typedef  unsigned long long ll;    
const int INF=1e9+100;     
const int mod=1e9+7;

int d[50][50][25][15],f[50];
int x,y,k,b;

int dfs(int pos,int num1,int sum,int dig,int limit,int lead){
    if(pos==-1){
        if(num1==sum) return 1;
        else return 0;
    }
    if(!lead&&!limit&&d[pos][num1][sum][dig]!=-1) return d[pos][num1][sum][dig];
    int up=limit?f[pos]:1;
    int ans=0;
    for(int i=0;i<=up&&i<=1;i++){
        if(i==1){
            ans+=dfs(pos-1,num1+1,sum,dig,limit&&i==up,0);
        }else{
            if(lead){
                ans+=dfs(pos-1,num1,sum,dig,limit&&i==up,1);
            }else{
                ans+=dfs(pos-1,num1,sum,dig,limit&&i==up,0);
            }
        }
    }
    if(!limit&&!lead) d[pos][num1][sum][dig]=ans;
    return ans;
}
int solve(int n){
    int pos=0;
    while(n){
        f[pos++]=n%b;
        n/=b;
    }
    return dfs(pos-1,0,k,b,1,1);
}
int main(){
    //freopen("out.txt","w",stdout);  
    memset(d,-1,sizeof(d));
    while(scanf("%d %d",&x,&y)!=EOF){
        scanf("%d %d",&k,&b);
        printf("%d\n",solve(y)-solve(x-1));
    }
    return 0;
}