一、度的数量
数 X X X 恰好等于 K K K 个互不相等的 B B B 的整数次幂之和即数 X X X 的 B B B 进制表示下有 K K K 位是 1 1 1,其余位均为 0 0 0。
数位DP的技巧:
- 题目要求区间 [ l , r ] [l,r] [l,r] 中满足性质的数的个数,用 d p ( n ) dp(n) dp(n) 求出区间 [ 0 , n ] [0,n] [0,n] 中满足性质的数的个数,则答案即为 d p ( r ) − d p ( l − 1 ) dp(r)-dp(l-1) dp(r)−dp(l−1)
- 按照树的形式,对区间 [ 0 , n ] [0,n] [0,n] 中的数从高到低位分解,一层一层处理 (如下图所示),一般每一层按照填 0 ~ a i − 1 0~a_i-1 0~ai−1 与填 a i a_i ai 分成两种情况,前者可以直接预处理求出结果,后者需要继续讨论下一位的选法。注意需要特判最后一层的右子节点是否满足性质。
设数 n n n 在 B B B 进制表示下有 m m m 位,第 i ( 0 ≤ i < m ) i\ (0\le i<m) i (0≤i<m) 位是 a i a_i ai (从低到高分别为第 0 0 0 位、第 1 1 1 位…第 m − 1 m-1 m−1 位),对于区间 [ 0 , n ] [0,n] [0,n] 中的数作以下分析:
当前处理到第 i i i 位时,且前所有位上共有 l a s t last last 个 1 1 1 时:
- a i > 0 a_i>0 ai>0,那么这一位可以填 0 0 0,剩下 i − 1 i-1 i−1 位还能填 K − l a s t K-last K−last 个 1 1 1,其余为 0 0 0,这样的方案数有 C i − 1 K − l a s t C_{i-1}^{K-last} Ci−1K−last
- a i > 1 a_i>1 ai>1,那么这一位可以填 1 1 1,剩下 i − 1 i-1 i−1 位还能填 K − l a s t − 1 K-last-1 K−last−1 个 1 1 1,其余为 0 0 0,这样的方案数有 C i − 1 K − l a s t − 1 C_{i-1}^{K-last-1} Ci−1K−last−1;此外,由于满足性质的数在 B B B 进制表示下只会有 0 0 0 和 1 1 1,因此这一位不能再填别的数了,直接返回
- a i = 1 a_i=1 ai=1,还需要对下一位继续讨论, l a s t + = 1 last+=1 last+=1
上述全部都是叶子节点为左子结点的情况,最后还要特判树的右支是否满足性质。
对于组合数 C n m C_n^m Cnm,我们可以根据递推公式 C n m = C n − 1 m + C n − 1 m − 1 C_n^m=C_{n-1}^m+C_{n-1}^{m-1} Cnm=Cn−1m+Cn−1m−1 ,通过两重循环预处理。
代码实现:
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int N = 35;
int 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