【题解】P2727 [USACO3.2] Stringsobits

题目链接(洛谷)

题目大意

考虑排好序的$N(1\le N \le 31)$位二进制数,且包含所有长度为$N$且这个二进制数中$1$的个数小于等于$L(l\le N)$的数。

现在输出满足长度为$N$,且$1$的个数小于等于$L$的第$k$小的那个二进制数。

分析

直接搜索全排列产生会超时。考虑$\text{dp}$做法。设$f(i,j)$表示前$i$位中,恰有$j$个$1$的二进制数量。容易得出$f(i,0)=1$, $f(i,j)=f(i-1,j)+f(i-1,j-1),~i$从$0$到$n$。求出在前$i$位中,最多有$j$个$1$的二进制数的数量,即$\sum_{j=0}^L f(i,j)$。如果这个数大于$k$,那么答案的第$i$位为$1$。

实现

$(41ms,~688.00kb)$

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#include<bits/stdc++.h>

#define ll long long
using namespace std;
const int N = 50;
ll n, l, k, dp[N][N];
bool ans[N];

void dfs(ll n, ll l, ll k) {
ll sum, lst;
for (int i = 0; i <= n; i++) {
lst = sum, sum = 0;
for (int j = 0; j <= l; j++) {
sum += dp[i][j];
if (sum >= k) {
ans[i] = true;
return dfs(n - 1, l - 1, k - lst);
}
}
}
}

int main() {
cin >> n >> l >> k;
for (int i = 0; i <= n; i++) {
dp[i][0] = 1;
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= i; j++) {
dp[i][j] = dp[i - 1][j] + dp[i - 1][j - 1];
}
}
dfs(n, l, k);
for (int i = n; i >= 1; i--) cout << ans[i];
return 0;
}