cf914c 暴力+组合（枚举1000）-优快云博客

本文链接：https://blog.youkuaiyun.com/c_czl/article/details/88392119

本文详细介绍了如何使用暴力枚举和组合数学的方法解决Codeforces比赛中的第914C题。通过分析问题，探讨了在限制计算复杂度为1000的情况下，如何巧妙地找出最优解。博客内容涵盖了算法设计、问题拆解及代码实现，对于提升编程竞赛解题能力具有指导意义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <map>

using namespace std;

long long mod=1e9+7;
char ch[1005];

long long dp[1055];

int k;

long long f(long long x)
{
    if(dp[x]!=-1)
        return dp[x];
    else
    {
        if(x==1)
            return dp[x]=0;
        else
        {
            long long cnt=0;
            while(x>0)
            {
                int cur=x%2;
                cnt+=cur;
                x/=2;
            }
            return dp[x]=f(cnt)+1;
        }
    }
}

long long fac[1055];
long long afac[1055];

long long powd(long long a,long long b)
{
    long long ans=1;
    while(b>0)
    {
        if(b%2==1)
            ans=ans*a%mod;
        a=a*a%mod;
        b/=2;
    }
    return ans;
}

long long C(long long b,long long a)
{
    if(a<0)
        return 0;
    if(a>b)
        return 0;
    if(a==b)
        return 1;
    if(a==0)
        return 1;

    return 1ll*(fac[b]*afac[b-a]%mod)*afac[a]%mod;
}

int main() {

    fac[0]=1;afac[0]=1;
    for(long long i=1;i<=1050;i++)
        fac[i]=fac[i-1]*i%mod,afac[i]=powd(fac[i],mod-2);

    for(int i=0;i<=1050;i++)
        dp[i]=-1;

    for(int i=1;i<=1050;i++)
    {
        dp[i]=f(i);
    }

    while(~scanf("%s",ch))
    {
        scanf("%d",&k);
        if(k==0)
        {
            printf("1\n");
        } else
        {
            int len=strlen(ch);
            long long ans=0;
            for(int i=1;i<=1005;i++)
            {
                int cur=i;
                if(dp[i]==k-1)
                {
                    for(int j=0;j<len;j++)
                    {
                        if(ch[j]=='1')
                        {
                            long long ans1=len-j-1;
                            ans=(ans+C(ans1,cur))%mod;
                            cur--;
                        }
                    }
                    if(cur==0)
                        ans=(ans+1)%mod;
                }
            }
            if(k==1)
                ans=(ans-1)%mod;
            ans=(ans+mod)%mod;
            printf("%lld\n",ans%mod);
        }
    }
    return 0;
}