Codeforces 914C

题目链接：http://codeforces.com/contest/914/problem/C

题面：

 

 题意：

给出一个数(二进制形式），每次将他变成他的二进制表示中1的个数， 问有多少个数可以正好在K此变化成为1。

题解：

首先用一个数组f记录下f[i]表示i这个数需要f[i]次变换成为1， 则有递推f[i] = f[__builtin_popcount(i)].

DP[i][j]表示前i位长度(下标从0开始), 有j个1的情况有多少种。

若当前位为0, 则DP[i][j] = DP[i-1][j]

若当前位为1，选的话就是DP[i-1][j-1]，不选就是C[i][j]; 则DP[i][j] = DP[i-1][j-1] + C[i][j];

 

#include <bits/stdc++.h>

using namespace std;
using ll = long long ;
const int MOD = 1E9 + 7;
const int maxn = 1 << 10;

int C[maxn][maxn];
int dp[maxn][maxn];
int f[maxn];
ll res;
void init()
{
    f[1] = 0;
    for(int i = 2; i < maxn; i ++) f[i] = f[__builtin_popcount(i)] + 1;
    C[0][0] = 1;
    for(int i = 1; i < maxn; i ++) {
        C[i][0] = 1;
        for(int j = 1; j <= i; j ++) {
            C[i][j] = C[i - 1][j - 1] + C[i - 1][j];
            C[i][j] %= MOD;
        }
    }
}

int main()
{
    ios::sync_with_stdio(false), cin.tie(0);
    init();
    int n, k;
    string s;
    cin >> s >> k;
    if(!k) {
        return cout << 1 << '\n', 0;
    }
    n = s.size();
    reverse(s.begin(), s.end());
    for(char & c : s) c -= '0';
    if(s[0]) {
        dp[0][1] = dp[0][0] = 1;
    } else {
        dp[0][0] = 1;
        dp[0][1] = 0;
    }
    for(int i = 1; i < n; i ++) {
        dp[i][0] = 1;
        for(int j = 1; j <= i + 1; j ++) {
            if(s[i]) {
                dp[i][j] = (dp[i - 1][j - 1] + C[i][j]) % MOD;
            } else {
                dp[i][j] = dp[i - 1][j] % MOD;
            }
        }
    }
    for(int i = 0; i <= n; i ++) {
        if(f[i] == k - 1) {
            res += dp[n - 1][i];
            res %= MOD;
        }
    }
    if(k == 1) res --;
    res = (res % MOD + MOD) % MOD;
    cout << res << '\n';
    return 0;
}