BZOJ 1037: [ZJOI2008]生日聚会Party

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

思路：记忆化搜索，我们一个一个位置考虑放男还是放女，用dp[n][m][k1][k2] 表示剩下n个男孩，m个女孩要分配，并且上一个位置往前延伸男孩最多多女孩k1个，女孩最多比男孩多-k2个的方案数

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string.h>
#include <queue>
#include <vector>
#include <algorithm>
#include <cassert>
#include <set>
#include <map>
#include <cmath>
#include <ctime>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<(b);++i)
#define rrep(i,a,b) for(int i=(a);i>=(b);--i)
#define clr(a,x) memset(a,(x),sizeof(a))
#define eps 1e-8
#define LL long long
#define mp make_pair
const int mod=12345678;
int dp[151][151][21][21];
int k;
int dfs(int n,int m,int k1,int k2)
{
    if(k1>k||k2<-k) return 0;
    if(k1<0) k1=0; if(k2>0) k2=0;
    if(dp[n][m][k1][k2+k]!=-1) return dp[n][m][k1][k2+k];
    int&d=dp[n][m][k1][k2+k];
    if(n==0&&m==0) return d=1;
    else if(n==0) return d=dfs(0,0,k1-m,k2-m);
    else if(m==0) return d=dfs(0,0,k1+n,k2+n);
    d=dfs(n-1,m,k1+1,k2+1);
    d+=dfs(n,m-1,k1-1,k2-1);
    if(d>mod) d-=mod;
    return d;
}

int main()
{
    int n,m;
    while(scanf("%d%d%d",&n,&m,&k)==3) {
        clr(dp,-1);
        printf("%d\n",dfs(n,m,0,0));
    }
}