HDU4301 Divide Chocolate DP问题

http://acm.hdu.edu.cn/showproblem.php?pid=4301

题目大意：给你一个2*n的巧克力，每一小块巧克力都长得不同，将其分成k块巧克力有多少种分发。

解题思路：用一个三维dp[i][j][k]，i表示正在解第i列，j表示已经被分成了j块，k表示i-1的俩块巧克力是否分开，0表连着，1表没有连着。

另外需要注意__int64可能会MLE，由于mod了一个数，仅仅相加的话不会爆int。

代码：

#include <math.h>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define MAXN 0
#define INF 0x7fffffff
#define eps 1e-10
#define mod 100000007
#define MAX(x,y) ((x)>(y)? (x):(y))
#define MIN(x,y) ((x)<(y)? (x):(y))
#define MEM(a) (memset((a),0,sizeof(a)))
#define FRE freopen("input.txt","r",stdin)
using namespace std;

int dp[1020][2040][2];

int main()
{
    int T;
    MEM(dp);
    dp[1][1][0]=dp[1][2][1]=1;
    for(int i=2;i<=1001;i++)
    {
        for(int j=1;j<=2002;j++)
        {
            if(j==1)
            {
                dp[i][j][0]=(dp[i-1][j][0]+2*dp[i-1][j][1]+dp[i-1][j-1][0]+dp[i-1][j-1][1])%mod;
                dp[i][j][1]=(2*dp[i-1][j-1][0]+dp[i-1][j][1]+2*dp[i-1][j-1][1])%mod;
            }
            else
            {
                dp[i][j][0]=(dp[i-1][j][0]+2*dp[i-1][j][1]+dp[i-1][j-1][0]+dp[i-1][j-1][1])%mod;
                dp[i][j][1]=(2*dp[i-1][j-1][0]+dp[i-1][j][1]+2*dp[i-1][j-1][1]+dp[i-1][j-2][0]+dp[i-1][j-2][1])%mod;
            }
        }
    }
    scanf("%d\n",&T);
    while(T--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        printf("%d\n",(dp[n][k][0]+dp[n][k][1])%mod);
    }
    return 0;
}