acwing1081. 度的数量（数位dp）

传送门

分析题意我们可以发现，我先最后就是看b进制下放k个1的合法方案数，详细看代码

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<set>
#include<map>
#include<bitset>
#include<unordered_map>
using namespace std;
#define LL long long
typedef unsigned long long ull;
#define eps (1e-9)
const int maxn = 40;
const int inf = 0x3f3f3f3f;
const int bas = 131;
const LL mod = 1e9 + 7;

int k, b;
int f[maxn][maxn];
void init()
{
	f[0][0] = 1;
	for (int i = 1; i <= 35; i++)
	{
		for (int j = 0; j <= 35; j++)
		{
			if (!j)f[i][j] = 1;
			else f[i][j] = f[i - 1][j] + f[i - 1][j - 1];//杨辉三角
		}
	}
}
int dp(int x)
{
	if (!x)return 0;
	vector<int>v;
	int res = 0, last = 0;
	while (x)
	{
		v.push_back(x % b);
		x /= b;
	}
	for (int i = v.size() - 1; i >= 0; i--)//从高位往低位判断
	{
		int tmp = v[i];
		if (tmp)
		{
			res += f[i][k - last];//这是当前位置0的情况
			if (tmp > 1)//当前数字大于1时，我们可以保证填1后整个数字还是小于x
			{
				if ((k - last - 1) >= 0)
				{
					res += f[i][k - last - 1];
				}
				break;//因为当前位置填1和不填1的情况都能算出来，后面的情况都已经算出来，所以直接结束
			}
			else//当前是1，这种情况
			{
				last++;//因为当前最大是1，0的情况已经算出来了，但1还算不出，因为此时后面的数不能随便放1，
				//有可能造成组合的数比x大，只能记录目前是1，也就是说我们往下找当前位置为1的合法方案，0已经
				//算出来了，所以往下遍历就好了。
				if (last > k)break;
			}
		}
		if (!i && last == k)res++; //比如1111这种情况，要在这特判一下。
	}
	return res;
}
int main()
{
	init();
	int x, y;
	scanf("%d%d%d%d", &x, &y, &k, &b);
	int ans = dp(y) - dp(x - 1);
	printf("%d", ans);
	return 0;
}