ural1057(数位dp)

链接：点击打开链接

题意：求给定区间[X,Y]中满足下列条件的整数个数：这个数恰好等于K个互不相等的B的整数次幂之和

代码：

#include <map>
#include <set>
#include <vector>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int siz=50;
int K,B;
int a[siz],dp[siz][siz][siz];
int dfs(int pos,int pre,int sta,int lim){
    int i,u;
    int ans=0;
    if(pos==0){
        if(sta==K)
        return 1;
        return 0;
    }
    if(dp[pos][pre][sta]!=-1&&lim==0)
    return dp[pos][pre][sta];
    u=lim?a[pos]:B-1;
    for(i=0;i<=u;i++){                          //只有考虑每一位的取值是1还是0
        if(i>1)                                 //是1的话sta+1
        break;
        else if(i==1)
        ans+=dfs(pos-1,i,sta+1,lim&(i==u));
        else
        ans+=dfs(pos-1,i,sta,lim&(i==u));
    }
    if(lim==0)
    dp[pos][pre][sta]=ans;
    return ans;
}
int cal(int x){
    int pos=0;
    while(x){
        a[++pos]=x%B;
        x/=B;
    }
    return dfs(pos,0,0,1);
}
int main(){
    int l,r;
    memset(dp,-1,sizeof(dp));
    while(scanf("%d%d",&l,&r)!=EOF){
        scanf("%d%d",&K,&B);
        printf("%d\n",cal(r)-cal(l-1));
    }
    return 0;
}
/*
15 20
2
2
*/