[USACO Section 3.2] 01串 Stringsobits (动态规划)

题目链接

Solution

贼有意思的 DP, 也可以用组合数学做.
(f[i][j]) 代表前 (i) 位,有 (j) 个 (1) 的方案数.
转移方程很简单 : (f[i][j]=f[i-1][j]+f[i-1][j-1])
然后可以按位判断答案上是否为 (1) .
如何判断?
如果当前 (f[i][L]) 小于 (I) ,那么我们所要的方案一定 (i) 位上为 (1);
同时用 (I) 减去 (f[i][L]) , (L) 同时减掉 (1);

Code

#include<bits/stdc++.h>
using namespace std;
int f[52][52]={0},ans[52];
int N, L;
long long I; 

int main()
{
    cin>>N>>L>>I;
    for (int i=0;i<=N;i++)f[i][0] =f[0][i]=1;
    
    for (int i=1;i<=N;i++)
        for (int j=1;j<=N;j++)
            if(j<=i)
            f[i][j]=f[i-1][j-1] + f[i-1][j];
            else f[i][j]=f[i][i];

    for (int i=N;i >=1;i--) 
    {
        if (I>f[i-1][L]) 
        {
            cout<<1;
            I-=f[i-1][L];
            L--;
        }
        else cout<<0 ;
    }
}