C. Travelling Salesman and Special Numbers

CF原题链接

题意：给定一个数 n，问 1 - n 中一共有多少数能够经过 k 次下面的转换变为 1，一次转换指的是将原来的数变为原来的数中二进制一个的个数转化为对应的十进制

思路：让我们将减少一个数到1所需的最小步数称为该数的阶数。
由于小于2^10000的数的设置位数（即二进制表示中1的个数）小于1000，因此任何小于2^10000的数都会在一步内减少到一个小于1000的数。我们可以预先计算前1000个数的阶数(使用ones函数)dp[1] = 0表示将1通过op变为1的操作次数为1。对于每个x（x<1000），如果x的阶数是k-1，我们需要计算小于或等于n的具有k个设置位的数的数量。
对于 n 的二进制每一位 1 来说，如果这一位 1 变为 0 了，那么修改过的数无论后面所有位是 0 是 1，都比原来的n小。
假设前面位置一共有 cnt 个1，后面还有n - i - 1个位置，那么就可以从后面的所有位置中也就是从n - i - 1 中选取x - cnt 就是组合数C[n - 1 - i][x - cnt]。

#include <bits/stdc++.h>
using namespace std;

#define rep(i, a, b) for (int i = a; i < b; i++)
#define per(i, a, b) for (int i = a; i >= b; i--)

const int N = 1e3 + 7, mod = 1e9 + 7;
int dp[N];
int64_t C[N][N];

inline int red() {
    int x;
    cin >> x;
    return x;
}

void init() {
    rep(i, 0, 1001) {
        C[i][0] = 1;
    }
    rep(i, 1, 1001) {
        rep(j, 1, 1001) {
            C[i][j] = (C[i - 1][j - 1] + C[i - 1][j]) % mod;
        }
    }
}

int ones(int n) {
    int res = 0;
    while (n) {
        if (n & 1) {
            res++;
        }
        n >>= 1;
    }
    return res;
}

void solve() {
    string s;
    int k;
    cin >> s >> k;
    init();
    int n = s.size();
    dp[1] = 0;
    rep(i, 2, 1001) {
        dp[i] = dp[ones(i)] + 1;
    }
    if (!k) {
        cout << "1\n";
        return ;
    }
    int64_t cnt = 0, res = 0;
    rep(i, 0, n) {
        if (s[i] == '0') {
            continue;
        }
        rep(j, max(cnt, 1ll), 1000) {
            if (dp[j] == k - 1) {
                res = (res + C[n - i - 1][j - cnt]) % mod;
            }
        }
        cnt++;
    }
    cnt = 0;
    for (char c : s) {
        if (c == '1') {
            cnt++;
        }
    }
    if (dp[cnt] == k - 1) {
        res = (res + 1) % mod;
    }
    if (k == 1) {
        res = (res + mod - 1) % mod;
    }
    cout << res << '\n';
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int t = 1;
    while (t--) {
        solve();
    }

    return 0; 
}