BZOJ 1037 生日聚会Party（DP）

题目链接：http://www.lydsy.com:808/JudgeOnline/problem.php?id=1037

题意:给定n个0m个1。有多少种排列方式使得任意一段中01的个数差小于等于K？

思路：f[i][j][k][t]表示i个0、j个1、0比1最多多k个、1比0最多多t的方案数。

 

int n,m,K,f[N][N][21][21];

void up(int &x,int y)
{
    x+=y;
    if(x>=mod) x-=mod;
}


int main()
{
    RD(n,m,K);
    int i,j,k,t;
    f[0][0][0][0]=1;
    FOR0(i,n+1) FOR0(j,m+1) FOR0(k,K+1) FOR0(t,K+1) if(f[i][j][k][t])
    {
        if(i<n&&k<K) up(f[i+1][j][k+1][max(t-1,0)],f[i][j][k][t]);
        if(j<m&&t<K) up(f[i][j+1][max(k-1,0)][t+1],f[i][j][k][t]);
    }
    int ans=0;
    FOR0(i,K+1) FOR0(j,K+1) up(ans,f[n][m][i][j]);
    PR(ans);
    return 0;
}