山东省第七届ACM F Feed the monkey (记忆搜索)

/*题意:3种水果,各有a,b,c个,一天喂一种水果,每种水果不能喂连续的n1,n2,n3天,问有多少种喂的方式
思路:简单的记忆搜索dp[i][j][k][s][f]表示第一种水果吃掉i个第二种水果吃掉j个，第三种水果剩余k个，以s种水果结尾连续数量 f 在规定范围内的组合种类*/
代码:
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const long long maxn = 52;
#define mod 1000000007
ll dp[maxn][maxn][maxn][4][maxn];
int N[4], D[4], sum;
int dfs(int a, int b, int c, int t, int d)
{
if (a > N[1] || b > N[2] || c > N[3] || a + b + c > sum) return 0;
for (int i = 1; i <= 3; ++i) {
if (t == i && d > D[i]) {
return 0;
}
}
if(dp[a][b][c][t][d] != -1) return dp[a][b][c][t][d];

if (a + b + c == sum) return 1;

int ans = 0;

if(t == 1) {
ans = (ans + dfs(a+1,b,c,1,d + 1)) % mod;
ans = (ans + dfs(a,b+1,c,2,1)) % mod;
ans = (ans + dfs(a,b,c+1,3,1)) % mod;
return dp[a][b][c][t][d]=ans;
}
if(t == 2) {
ans = (ans + dfs(a+1,b,c,1,1)) % mod;
ans = (ans + dfs(a,b+1,c,2,d + 1)) % mod;
ans = (ans + dfs(a,b,c+1,3,1)) % mod;
return dp[a][b][c][t][d]=ans;
}
if(t == 3) {
ans = (ans + dfs(a+1,b,c,1,1)) % mod;
ans = (ans + dfs(a,b+1,c,2,1)) % mod;
ans = (ans + dfs(a,b,c+1,3,d + 1)) % mod;
return dp[a][b][c][t][d]=ans;
}
}
int main()
{
int Test;
scanf("%d", &Test);
while (Test--) {
memset(dp, -1, sizeof(dp));
scanf("%d%d%d%d%d%d", &N[1], &N[2], &N[3], &D[1], &D[2], &D[3]);
int ans = 0;
sum = N[1] + N[2] + N[3];
ans = (ans + dfs(1, 0, 0, 1, 1)) % mod;
// cout << ans << endl;
ans = (ans + dfs(0, 1, 0, 2, 1)) % mod;
// cout << ans << endl;
ans = (ans + dfs(0, 0, 1, 3, 1)) % mod;
cout << ans << endl;
}
return 0;
}