BZOJ1037: [ZJOI2008]生日聚会Party

DP…

/**************************************************************
    Problem: 1037
    User: zhuohan123
    Language: C++
    Result: Accepted
    Time:448 ms
    Memory:1792 kb
****************************************************************/
 
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
inline int imax(int a,int b){return a>b?a:b;}
inline int imin(int a,int b){return a<b?a:b;}
const int mo=12345678;
inline int add(int &a,int b){a=(a+b)%mo;}
int f[2][151][21][21];
//f[总人数][男生数][男-女][女-男] P.S.最后两维若是负数则当为0 
int main(int argc, char *argv[])
{
    int n,m,k;cin>>n>>m>>k;
    f[0][0][0][0]=1;
    for(int i=0;i<n+m;i++)
    {
        memset(f[(i+1)&1],0,sizeof f[(i+1)&1]);
        for(int j=0;j<=imin(i,n);j++)
            for(int bmg=0;bmg<=imin(j,k);bmg++)
                for(int gmb=0;gmb<=imin(i-j,k);gmb++)
                {
                    if(bmg<k&&j<n)add(f[(i+1)&1][j+1][bmg+1][imax(gmb-1,0)],f[i&1][j][bmg][gmb]);
                    if(gmb<k&&(i-j)<m)add(f[(i+1)&1][j][imax(bmg-1,0)][gmb+1],f[i&1][j][bmg][gmb]);
                }
    }   
    int ans=0;
    for(int i=0;i<=k;i++)
        for(int j=0;j<=k;j++)
        add(ans,f[(n+m)&1][n][i][j]);
    cout<<ans<<endl;
    return 0;
}