HDU1545（01-K Code）

最新推荐文章于 2020-05-29 10:30:00 发布

huanzhizun

最新推荐文章于 2020-05-29 10:30:00 发布

阅读量1.2k

点赞数

分类专栏：动态规划文章标签： ACM 动态规划

本文链接：https://blog.csdn.net/u013665921/article/details/38236661

版权

动态规划专栏收录该内容

45 篇文章 0 订阅

订阅专栏

dp[i][j][k]=dp[i-1][j+1][k-1]+dp[i-1][j-1][k+1].

dp[i][j][k]表示前i个1比0最多多j，0比1最多多k，注意是连续从1开始。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define LL __int64
using namespace std;
LL dp[100][10][10];
void dfs(int n,int m){
    dp[1][0][1]=1;
    dp[1][1][0]=1;
    int i,j,k;
    for(i=2;i<=n;i++){
        for(j=1;j<=m;j++){
            dp[i][0][j]=dp[i-1][0][j-1]+dp[i-1][1][j-1];
            dp[i][j][0]=dp[i-1][j-1][0]+dp[i-1][j-1][1];
        }
        for(j=1;j<m;j++){
            for(k=1;k<m;k++){
                dp[i][j][k]=dp[i-1][j-1][k+1]+dp[i-1][j+1][k-1];
            }
        }
        for(j=1;j<m;j++){
            dp[i][j][m]=dp[i-1][j+1][m-1];
            dp[i][m][j]=dp[i-1][m-1][j+1];
        }
    }
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        dfs(n,m);
        LL p=0;
        for(int i=0;i<=m;i++){
            for(int j=0;j<=m;j++){
                p+=dp[n][i][j];
            }
        }
        printf("%I64d\n",p);
    }
}