CodeForces 431 D.Random Task（组合数学）

Description

求满足$n+1$~$2n$之间恰有$m$个数二进制表示有$k$个$1$的$n$

Input

两个正整数$m,k(0\le m\le 10^{18},1\le k\le 64)$

Output

保证有解，输出满足条件的$n$，多组解输出一个即可

Sample Input

1 1

Sample Output

1

Solution

令$num[n]$为$n+1$~$2n$之间二进制表示为$k$的数的个数，由于$2n$二进制表示$1$的个数和$n$的相同，故有$num[n]=num[n-1]+check(2n-1)$，其中$check(2n-1)=1$，如果$2n-1$二进制表示有$k$个$1$，否则为$0$，进而有$num[n]=check(2n-1)+check(2n-3)+...+check(1)$

如果$k=1$，那只有$1$二进制表示为$1$个$1$，故此时答案为$0$

如果$k\ge 2$，那么$check(1)=0$，$num[n]=check(3)+...+check(2n-1)$，注意到这里面都是奇数，其最高位和最低位都是$1$，对于二进制长度为$x$的奇数，其二进制表示中有$k$个$1$的有$C_{x-2}^{k-2}$个数，从小到大枚举$x$，如果$m>C_{x-2}^{k-2}$，说明$2n-1$的长度必然大于$x$，从$m$中把这个长度的数减掉，直至找到$x$满足$m\le C_{x-2}^{k-2}$，说明$2n-1$的二进制表示长度为$x$

如果$2n-1$的次高位取$0$，那么剩下的$x-3$位要放$k-2$个$1$，方案数$C_{x-3}^{k-2}$，如果$m>C_{x-3}^{k-2}$说明这一位不能是$0$，否则不够$m$个数字，那这一位只能是$1$，然后从$m$中减掉这一位是$0$的方案数，否则说明这一位是$0$，以此从高位到低位确定$2n-1$每一位的取值， 进而得到$2n-1$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
ll C[66][66];
void init(int n=62)
{
    C[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;j++)C[i][j]=C[i-1][j-1]+C[i-1][j];
    }
}
int main()
{
    init();
    ll m;
    int k;
    while(~scanf("%I64d%d",&m,&k))
    {
        if(k==1||m==0)
        {
            printf("1\n");
            continue;
        }
        int n=k;
        while(1)
        {
            if(C[n-2][k-2]>=m)break;
            m-=C[n-2][k-2];
            n++;
        }
        ll ans=(1ll<<(n-1))+1;
        k-=2;
        for(int i=n-2;i&&k;i--)
        {
            if(m>C[i-1][k])
                m-=C[i-1][k],k--,ans+=(1ll<<i);
        }
        printf("%I64d\n",(ans+1)>>1);
    }
    return 0;
}