AcWing 5271易变数

akeyy

已于 2023-11-15 21:57:10 修改

阅读量37

点赞数

分类专栏： AcWing算法题解文章标签：算法动态规划

于 2023-11-08 21:35:21 首次发布

本文链接：https://blog.csdn.net/akeyy/article/details/134298567

版权

AcWing算法题解专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题目描述

原题链接：5271.易变数

思路

1.先特判一些输入的n,k

当 $k = 0$ 的时候，因为只有1操作0次为1，而且不管n输入多少，区间总会包含1，因此直接输出1即可。
当 $n = 1$ ， $k\neq0$ 的时候，直接输出0即可。

2.求解g[i]

假设 $g [i]$ 代表 $i$ 这个数操作多少次恰好为1。设 $c n t$ 为 $i$ 的二进制数中1的个数。
那么 $g [i] = g [c n t] + 1$ 。
例如： $g[4]=g[100_{(2)}]=g[1]+1=1$ 。
根据上述公式，我们可以求出 $[1, 1000]$ 的 $g [i]$ 的值。

3.求解组合数

$C_{i}^{j}$ 表示在 $i$ 个物品中选 $j$ 个物品的方案数
如果考虑第 $i$ 个物品选还是不选，我们就可以有两种情况：
选：那么方案数就是从剩下的 $i - 1$ 个物品中选 $j - 1$ 个，即 $C_{i-1}^{j-1}$
不选：那么方案数就是从剩下的 $i - 1$ 个物品中选 $j$ 个，即 $C_{i-1}^{j}$
那么， $C_{i}^{j} = C_{i-1}^{j-1}+C_{i-1}^{j}$

4.树形枚举

在这里插入图片描述
在枚举的时候，如果我们从1开始一个一个枚举，那么时间复杂度就是 $o(2^{1000})$ 。但是我们其实并不关心这个数具体是什么，我们只关心它的一些性质，比如这个数的二进制有多少个1，因此这个时候就可以用树形枚举。

假设一个二进制数1101，我们可以从它的最高位开始枚举，假如当前位为1，那么这一位就有两种可能，如果为0，就只有1种可能。（因为如果枚举这一位为1的话就会超过 $n$ ）

假如当前位为1且我们枚举的时候让其为0，那么后面的数我们就可以随便填而不会超过 $n$ 了。

我们用 $t$ 存储当前这一位之前的所有1的个数，枚举后面所有可能填数的情况，这个时候我们之前算的组合数就用上了。用 $j$ 表示后面可以随便填的数中1的个数， $c n t$ 表示可以随便填的位数，那么整个数的1的个数就是 $t + j$ ，这个时候我们只需要判断一下 $g [t + c n t] == k - 1$ 即可，如果相等，那么 $res +=C_{cnt}^{j}$ 。

最后不要忘记把 $t$ 递增。

C++代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int P = 1e9+7;
const int N = 1010;
string s;
int k, g[N], c[N][N]; // g[i]记录i到1需要多少次操作,c[i][j]表示组合数
int main()
{
    cin >> s >> k;
    if(k == 0)
    {
        cout << 1;
        return 0;
    }
    else if(s.size() == 1)
    {
        cout << 0;
        return 0;
    }
    g[0] = -1;
    for(int i = 2; i <= N; i++)
    {
        int x = i;
        int cnt = 0; // 记录i有多少个1
        while(x)
        {
            if(x & 1) cnt++;
            x >>= 1;
        }
        g[i] = g[cnt]+1;
    }
    for(int i = 0; i <= 1000; i++) c[i][0] = 1; 
    for(int i = 1; i <= 1000; i++)
    {
        for(int j = 0; j <= i; j++)
        {
            c[i][j] = c[i-1][j];
            if(j > 0) c[i][j] = (c[i][j] + c[i-1][j-1]) % P;
        }
    }
    int n = s.size(), t = 0, res = 0; // t表示前i+1个数里有多少个1
    for(int i = 0; i < n; i++)
    {
        if(s[i] == '1')
        {
            int cnt = n-i-1; // 表示可以随便填的数的个数
            for(int j = 0; j <= cnt; j++) // j表示后面cnt个里有j个1
            {
                if(g[t+j] == k-1) // 如果这个数经过k次操作可以变为1
                {
                    if(t+j == 1 && j == 1)
                    {
                        res = (res + c[cnt][j] - 1) % P; // 1变为1的操作个数为0, 例如100,010,001，表面上是三种，实际上要减去001
                    }
                    else 
                    {
                        res = (res + c[cnt][j]) % P;
                    }
                }
            }
            t++;
        }
    }
    if(g[t] == k-1) res = (res + 1) % P;
    cout << res << endl;
}