BZOJ 1037: [ZJOI2008]生日聚会Party 四维DP

1037: [ZJOI2008]生日聚会Party

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1650  Solved: 971
[Submit][Status][Discuss]

Description

今天是hidadz小朋友的生日，她邀请了许多朋友来参加她的生日party。 hidadz带着朋友们来到花园中，打算坐成一排玩游戏。为了游戏不至于无聊，就座的方案应满足如下条件：对于任意连续的一段，男孩与女孩的数目之差不超过k。很快，小朋友便找到了一种方案坐了下来开始游戏。hidadz的好朋友Susie发现，这样的就座方案其实是很多的，所以大家很快就找到了一种，那么到底有多少种呢？热爱数学的hidadz和她的朋友们开始思考这个问题…… 假设参加party的人中共有n个男孩与m个女孩，你是否能解答Susie和hidadz的疑问呢？由于这个数目可能很多，他们只想知道这个数目除以12345678的余数。

Input

仅包含一行共3个整数，分别为男孩数目n, 女孩数目m, 常数k。

Output

应包含一行，为题中要求的答案。

Sample Input

1 2 1

Sample Output

1

HINT

对于100%的数据, n , m ≤ 150，k ≤ 20。

题解：dp[i][j][k1][k2]表示i男，j女时，男比女最多多k1个，女比男最多多k2个的方案数

///

#include<iostream>

#include<cstdio>

#include<cstring>

#include<string>

#include<algorithm>

#include<queue>

#include<cmath>

#include<map>

#include<bitset>

#include<set>

#include<vector>

using namespace std ;

typedef long long ll;

#define mem(a) memset(a,0,sizeof(a))

#define meminf(a) memset(a,127,sizeof(a))

#define memfy(a) memset(a,-1,sizeof(a))

#define TS printf("111111\n")

#define FOR(i,a,b) for( int i=a;i<=b;i++)

#define FORJ(i,a,b) for(int i=a;i>=b;i--)

#define READ(a,b) scanf("%d%d",&a,&b)

#define g 9.8

#define mod 12345678

#define eps 1e-7

#define maxn  1000000+10

#define inf 10000000000;

inline ll read()

{

    ll x=,f=;

    char ch=getchar();

    while(ch<''||ch>'')

    {

        if(ch=='-')f=-;

        ch=getchar();

    }

    while(ch>=''&&ch<='')

    {

        x=x*+ch-'';

        ch=getchar();

    }

    return x*f;

}

//****************************************

int dp[][][][];

int main()

{

    mem(dp);

    int n=read(),m=read(),k=read();dp[][][][]=;

    FOR(i,,n)FOR(j,,m)FOR(k1,,k)FOR(k2,,k){

       dp[i+][j][k1+][max(k2-,)]+=dp[i][j][k1][k2];dp[i+][j][k1+][max(k2-,)]%=mod;

       dp[i][j+][max(k1-,)][k2+]+=dp[i][j][k1][k2];dp[i][j+][max(k1-,)][k2+]%=mod;

    }ll ans=;

    FOR(i,,k)FOR(j,,k){ans+=dp[n][m][i][j];ans%=mod;}

    cout<<ans<<endl;

    return ;

}

代码