SCAU周训4-F：URAL - 1057

1.题目描述：

2.题意：
给你一个区间[X,Y]，计数区间中所有由K个不同的B的幂的和组成的数字的个数。

3.思路：
数位dp。 题意说白了：计数B进制下，数字由K个1其余都是0组成的数字的个数。

4.代码：

//F
#include
#define rep(i,a,b) for(int i=(a);i<(b);++i)
#define mem(a,b) memset(a,(b),sizeof(a))
using namespace std;

typedef long long ll;

int l,r,k,b;
int dig[50];
int dp[50][50];

int dfs(int pos,int cnt,bool lim)
{
	if(!pos) return cnt==k;
	if(!lim&&~dp[pos][cnt]) return dp[pos][cnt];
	int up=lim?(dig[pos]?1:0):1,res=0;
	rep(i,0,up+1) res+=dfs(pos-1,cnt+i,lim&&i==dig[pos]);
	if(!lim) dp[pos][cnt]=res;
	return res;
}

int work(int n)
{
	int pos=0;
	while(n) dig[++pos]=n%b,n/=b;
	return dfs(pos,0,1);
}

inline void solve()
{
	mem(dp,-1);
	printf("%d",work(r)-work(l-1));
}

int main()
{
	while(~scanf("%d%d%d%d",&l,&r,&k,&b)) solve();
	return 0;
}