AcWing 1081. 度的数量 简单明了解释
原题链接
中等
作者:
int_main
,
2022-06-24 12:54:10
,
所有人可见
,
阅读 158
核心思想代码注释写得很清楚
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 35;
int x, y, k, b;
int f[N][N];
void init()
{
for (int i = 0; i < N; i ++)
for (int j = 0; j <= i; j ++)
if (!j) f[i][j] = 1;
else f[i][j] = f[i - 1][j] + f[i - 1][j - 1];
}
int dp(int y)
{
if (!y) return 0;
int ans = 0, last = 0;
vector<int> nums;
while (y) nums.push_back(y % b), y /= b;
for (int i = nums.size() - 1; i >= 0; i --)
{
int x = nums[i];
if (x)
{
ans += f[i][k - last];
//当前位大于0
//那么选择0,以后随便选
if (x > 1)
{
if (k - last - 1 >= 0) ans += f[i][k - last - 1];
break;
//如果当前位大于1,那么以后的可以随便选,选完就break
}
else
{
last ++;
if (last > k) break;
//如果当前位==1
//那么把可用的1减少,往下看
}
}
if (!i && k == last) ans ++;
//如果最后一位等于0
//选0,答案+1
}
return ans;
}
int main()
{
cin >> x >> y >> k >> b;
init();
cout << dp(y) - dp(x - 1);
}