您的位置:首页 > 其它

DP(比基础稍难练习?)

2016-06-14 22:48 204 查看
http://acm.sdut.edu.cn/sdutoj/showproblem.php?pid=3565&cid=1761

题意:共有三种水果,每种水果都有一定的数量Ni,每种水果都不能连续Di天出现。问有多少种组合方式?

T组数据,每组数据6个数依次是三种水果的数量,和三种水果不能连续出现的天数

分析:首先需要明确这是动态规划。且是五维的,前三维为各个水果的已经使用的数量,第四维是目前该种水果已连续出现多少天,第五维是当前水果的种类。

使用记忆化搜索的方法。

因为有当前水果这一维度,就需要对进入搜索时是哪种水果进行枚举。此时需判断连续出现的天数不能少于1天

搜索时要判断两点一是数量超不超,二是连续的长度超不超

//注意dp维数较多时,开的dp的大小要和用的时候对应
//因为开的dp维数的对应不对,屡wa不止
#include <iostream>
#include <cstdio>
#include <cstring>
#include <sstream>
#include <string>
#include <algorithm>
#include <list>
#include <map>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <cstdlib>
using namespace std;
int const maxn = 55;
int n1,n2,n3,d1,d2,d3;
int const mod = 1000000007;
int dp[maxn][maxn][maxn][maxn][3];
int dfs(int n,int a,int b,int c,int num,int longg)
{

if(n == n1 + n2 + n3 )
return 1;
if(dp[a][b][c][longg][num] != -1)
return dp[a][b][c][longg][num];
dp[a][b][c][longg][num] = 0;
int t = dp[a][b][c][longg][num];
if(num == 0)//当前的水果味第一种水果
{
if(d1>= longg + 1 && n1 >=a + 1)//当前水果的个数不超,当前连续的水果数不超
t=(t + dfs(n + 1,a + 1,b,c,0,longg + 1)) % mod;
if(n2 >= b + 1 && d2 > 0)//第二种水果的个数不超,且第二种水果的连续个数要大于0
t=( t + dfs(n + 1,a ,b + 1,c,1, 1)) % mod;
if(n3>= c + 1&& d3> 0)
t= (t + dfs(n + 1,a ,b, c + 1,2, 1)) % mod;
}
else if(num == 1)
{
if(n1 >= a + 1&& d1 > 0)
t=( t + dfs(n + 1,a + 1,b,c,0,1)) % mod;
if(d2 >= longg + 1 && n2 >= b + 1)
t= (t + dfs(n + 1,a ,b + 1,c,1, longg + 1) ) % mod;
if(n3>= c + 1&& d3 > 0)
t=( t + dfs(n + 1,a ,b, c + 1,2, 1) ) % mod;
}
else
{
if( n1 >= a + 1&& d1 > 0)
t= (t + dfs(n + 1,a + 1,b,c,0,1)) % mod;
if(n2 >= b + 1&& d2 > 0)
t= (t + dfs(n + 1,a ,b + 1,c,1, 1)) % mod;
if(d3 >= longg + 1&&n3>= c + 1)
t = (t + dfs(n + 1,a ,b, c + 1,2, longg + 1)) % mod;
}
dp[a][b][c][longg][num] = t% mod;
return dp[a][b][c][longg][num];
}
int main()
{
// freopen("in.txt","r",stdin);
int T;
scanf("%d",&T);
while(T--)
{

memset(dp,-1,sizeof(dp));
scanf("%d%d%d%d%d%d",&n1,&n2,&n3,&d1,&d2,&d3);
int ans = 0;
if(d1 > 0)//第一种水果的个数大于1,另当前水果为1,第一种水果的个数-1
ans = (ans + dfs(1,1,0,0,0,1)) % mod ;
if(d2 > 0)
ans= (ans + dfs(1,0,1,0,1,1)) % mod;
if(d3 > 0)
ans =(ans + dfs(1,0,0,1,2,1 )) % mod;
cout<<ans<<endl;
}
return 0;
}

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  acm 算法 dp