蓝桥杯 国赛 2021

蓝桥杯
国赛2021
二进制问题
B组

dp[0][i][j]不超过前i位的数字取j个1
dp[1][i][j]超过前i位的数字取j个1
对第i位取0和1分情况计算转移

#include<iostream>
using namespace std;
int xl[100],cl;
long long dp[2][100][100];
int main()
{
    long long n;
    int k;
    cin>>n>>k;
    while(n)
    {
        xl[++cl]=n%2;
        n/=2;
    }
    for(int i=0;i<=cl;i++) dp[0][i][0]=1;
    for(int i=1;i<=cl;i++)
    {
        for(int j=1;j<=min(i,k);j++)
        {
            if(xl[i]==0)
            {
                dp[0][i][j]=dp[0][i-1][j];
                dp[1][i][j]=dp[1][i-1][j]+dp[0][i-1][j-1]+dp[1][i-1][j-1];
            }
            else
            {
                dp[0][i][j]=dp[0][i-1][j]+dp[1][i-1][j]+dp[0][i-1][j-1];
                dp[1][i][j]=dp[1][i-1][j-1];
            }
        }
    }
    cout<<dp[0][cl][k];
}