思维 + 二进制字符串处理：Strange Memory

最新推荐文章于 2024-09-13 15:19:30 发布

啦啦啦32421

最新推荐文章于 2024-09-13 15:19:30 发布

阅读量108

点赞数

分类专栏：基础算法文章标签：模拟排列组合

本文链接：https://blog.csdn.net/qq_49120553/article/details/121621224

版权

基础算法专栏收录该内容

111 篇文章 0 订阅

订阅专栏

该博客详细解析了Codeforces Problem D的题目，并提供了O(N)复杂度的解决方案。通过枚举二进制串中1的个数，利用动态规划计算不同1个数的组合数，并结合题目给定的常数c，求得最终答案。博客还给出了完整的C++代码实现，包括前缀和计算和动态规划表格初始化。

摘要由CSDN通过智能技术生成

题目链接：Problem - D - Codeforces

分析：

此题枚举一下就会发现：下标i中对应的1的个数有多少个(如x个)，c[]中最大值就为c ^ x。

所以此题最主要的就是求从1 ~ n中二进制下遍历统计所有情况的总个数。

比如二进制下只有一个1的总个数，只有2个1的总个数，只有3个1的总个数......

而这个的求法，可以使用O(N)的方式进行求解。

参考文献：https://blog.csdn.net/qq_49120553/article/details/121621112

代码实现：

# include <iostream>
# include <string>
# include <cstring>
using namespace std;

const int N = 3010 , mod = 1e9 + 7;

int a[N]; // 统计对应的1有多少个

char ch[N];

long long t[N][N];

int sum[N]; // 前缀和

long long c;

long long tc[N];

void init(int n)
{
    t[0][0] = 1;
    for(int i = 1 ; i <= n ; i++)
    {
        for(int j = 0 ; j <= i ; j++)
        {
            if(!j)
            {
                t[i][j] = 1;
            }
            else
            {
                t[i][j] = ( t[i - 1][j - 1] + t[i - 1][j] ) % mod;
            }
        }
    }
}

int main()
{
    scanf("%s %lld",ch + 1 , &c);

    bool flag = false;
    for(int i = 1 ; ch[i] ; i++)
    {
        if(ch[i] == '1')
        {
            flag = true;
            break;
        }
    }
    if(!flag)
    {
        printf("%d\n",1);
        return 0;
    }

    int len  = strlen(ch +1); // 长度


    init(len + 5);

    tc[0] = 1;
    for(int i = 1 ; i <= len ; i++)
    {
        tc[i] = ( tc[i - 1] * c ) % mod;
    }

    for(int i = 1 ; i <= len ; i++)
    {
        sum[i] = sum[i - 1] + (ch[i] == '1');
    }

    for(int i = len ; i >= 1 ; i--)
    {
        if(ch[i] == '0')
        {
            continue;
        }

        // 1 ~ i - 1中目前1的个数 , 因为会将i的1变为0，后面的任意选择

        int num2 = sum[i - 1] - sum[0];

        // i + 1 到 len共有多少位
        int templen = len - (i + 1) + 1;

        for(int j = 0 ; j <= templen ; j++) // i这个位置变为0 ，j个1
        {
            a[num2 + j ] = (a[num2 + j ] + t[templen][j]) % mod ;
        }
    }
    a[sum[len] - sum[0] ]++;


    // 最多会有len个1
    long long ans = 0;


    for(int i = 0 ; i <= len ; i++)
    {
        ans = ( ans + ( a[i] * tc[i] ) ) % mod;
    }
    // 在最高位的地方为1的时候，会算一次全0的情况，因为会将最高位的1先变为0，然后后面的随便排
    printf("%lld\n",ans);
    return 0;
}