题意:输出第n个不含前导0且1的个数为k的二进制串。
这题和之前一道题类似:G:aab aba baa
k=1的情况比较容易考虑,对于任意的n只要在1后面添加若干个零。
对于k>1的情况,先把首位的1拿出来,再讨论剩余k-1个1的分布。
可以枚举串长度为k,k+1,k+2,k+3… L。要解决这个题,首先要知道
这个串的长度,也就是说至少多少位01才能表示出第n种情况。
直接判断前面组合数的前缀和,直到某个前缀不小于n。
#include <iostream>
#include <vector>
#define int unsigned long long
using namespace std;
int ask(int a,int b)
{
if(b==1)return a;
if(b==2)return a * (a-1) / 2 ;
int ans = 1 ;
for(int i = a ; i >= a - b + 1 ; i--){
ans = ans * i ;
}
for(int i = b ; i >= 2 ; i--){
ans = ans / i ;
}
return ans ;
}
vector<char>ans;
signed main()
{
//freopen("in","r",stdin);
int n , k ;
cin >> n >> k ;
if(k==1){
putchar('1');
for(int i = 1 ; i < n ; i++) putchar('0');
}else{
int L = 0 ;
int sum = 0 ;
int lassum = 0 ;
for(int i = k - 1 ; i ; i++){
sum = sum + ask(i,k-1);
if(sum >= n){
L = i ;break;
}
lassum = sum ;
}
n = n - lassum ;
ans.push_back('1');
k--;
for(int i = L ; i >= 1 ; i--){
int num1 = ask(i - 1,k);
if(num1 < n){
n = n - num1 ;
ans.push_back('1');
k--;
}else{
ans.push_back('0');
}
}
for(int i = 0 ; i < ans.size() ; i++){
putchar(ans[i]);
}
}
return 0;
}