1037: [ZJOI2008]生日聚会Party

o(╯□╰)o看了题解才会。。所以总结一下吧。。。

这道题是dp或者说是递推。

显然m,n要放到状态里，考虑直接用题目的答案来推答案。

问题是要保证相差个数不超过k来进行转移，这里转移的时候是用前面的来推后面的。

一开始想用f[i,j,k]表示i个男孩j个女孩男孩比女孩最多多k的方案数量，但是就无法记录女孩比男孩多的情况。

设dp[i][j][x1][x2]表示i个男孩j个女孩

男孩最多比女孩多x1个

女孩最多比男孩多x2个

时的方案种数。

已知f[0][0][0][0]=1

那么

dp[i+1][j][x1+1][max(x2-1,0)]+=dp[i][j][x1][x2]) (max是防止负数的情况)

同理可以写出另一个方程来。

#include<iostream>
#include<algorithm>
using namespace std;
const int M=12345678;
int dp[155][155][25][25];
int main()
{
    int n,m,k;
    long long ans=0;
    cin>>n>>m>>k;
    dp[0][0][0][0]=1;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=m;j++)
            for(int x1=0;x1<=k;x1++)
                for(int x2=0;x2<=k;x2++)
                {
                    dp[i+1][j][x1+1][max(x2-1,0)]=((long long)dp[i+1][j][x1+1][max(x2-1,0)]+dp[i][j][x1][x2])%M;
                    dp[i][j+1][max(x1-1,0)][x2+1]=((long long)dp[i][j+1][max(x1-1,0)][x2+1]+dp[i][j][x1][x2])%M;
                }
    for(int x1=0;x1<=k;x1++)
        for(int x2=0;x2<=k;x2++)
        ans=(ans+dp[n][m][x1][x2])%M;
    cout<<ans;
    while(1);
    return 0;
}