BZOJ1037: [ZJOI2008]生日聚会Party

DP…

 1 /**************************************************************

 2     Problem: 1037

 3     User: zhuohan123

 4     Language: C++

 5     Result: Accepted

 6     Time:448 ms

 7     Memory:1792 kb

 8 ****************************************************************/

 9  

10 #include <iostream>

11 #include <cstdio>

12 #include <cstring>

13 using namespace std;

14 inline int imax(int a,int b){return a>b?a:b;}

15 inline int imin(int a,int b){return a<b?a:b;}

16 const int mo=12345678;

17 inline int add(int &a,int b){a=(a+b)%mo;}

18 int f[2][151][21][21];

19 //f[总人数][男生数][男-女][女-男] P.S.最后两维若是负数则当为0 

20 int main(int argc, char *argv[])

21 {

22     int n,m,k;cin>>n>>m>>k;

23     f[0][0][0][0]=1;

24     for(int i=0;i<n+m;i++)

25     {

26         memset(f[(i+1)&1],0,sizeof f[(i+1)&1]);

27         for(int j=0;j<=imin(i,n);j++)

28             for(int bmg=0;bmg<=imin(j,k);bmg++)

29                 for(int gmb=0;gmb<=imin(i-j,k);gmb++)

30                 {

31                     if(bmg<k&&j<n)add(f[(i+1)&1][j+1][bmg+1][imax(gmb-1,0)],f[i&1][j][bmg][gmb]);

32                     if(gmb<k&&(i-j)<m)add(f[(i+1)&1][j][imax(bmg-1,0)][gmb+1],f[i&1][j][bmg][gmb]);

33                 }

34     }   

35     int ans=0;

36     for(int i=0;i<=k;i++)

37         for(int j=0;j<=k;j++)

38         add(ans,f[(n+m)&1][n][i][j]);

39     cout<<ans<<endl;

40     return 0;

41 }