1037: [ZJOI2008]生日聚会Party
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 755 Solved: 453
[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
对于30%的数据,n , m ≤ 20;
对于100%的数据, n , m ≤ 150,k ≤ 20。
解析:
比较容易看出是DP,但状态不太好想。。。
dp[i][j][k1][k2]表示前i个人中,有j个男孩,男孩最多比女孩多k1个,女孩最多比男孩多k2个的最大方案数。。。
如果第i+1个是男孩:dp[i+1][j+1][k1+1][k2-1]=sigma(dp[i][j][k1][k2])
第i+1个是女孩:dp[i+1][j][k1-1][k2+1]=sigma(dp[i][j][k1][k2])
因为四维存不下,而且状态i+1只与状态i有关,所以要用01滚动才能过。。。
代码:
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,k,dp[2][151][21][21];
int mod=12345678;
void read()
{
freopen("party.in","r",stdin);
freopen("party.out","w",stdout);
}
void work()
{
scanf("%d%d%d",&n,&m,&k);
int cs=1,ns=0;
dp[0][0][0][0]=1;
for(int i=0;i<n+m;i++)
{
swap(cs,ns);
memset(dp[ns],0,sizeof(dp[ns]));
for(int j=0;j<=min(i,n);j++)
for(int k1=0;k1<=min(j,k);k1++)
for(int k2=0;k2<=min(i-j,k);k2++)
{
if(k1<k && j<n) (dp[ns][j+1][k1+1][max(k2-1,0)]%=mod)+=(dp[cs][j][k1][k2]%=mod);
if(k2<k && i-j<m) (dp[ns][j][max(k1-1,0)][k2+1]%=mod)+=(dp[cs][j][k1][k2]%=mod);
}
}
int ans=0;
for(int k1=0;k1<=k;k1++)
for(int k2=0;k2<=k;k2++)
(ans+=(dp[ns][n][k1][k2]%=mod))%=mod;;
printf("%d\n",ans);
}
int main()
{
read();
work();
return 0;
}