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;
}
题意:共有三种水果,每种水果都有一定的数量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;
}
相关文章推荐
- 书评:《算法之美( Algorithms to Live By )》
- 动易2006序列号破解算法公布
- Ruby实现的矩阵连乘算法
- C#插入法排序算法实例分析
- 超大数据量存储常用数据库分表分库算法总结
- C#数据结构与算法揭秘二
- C#冒泡法排序算法实例分析
- 算法练习之从String.indexOf的模拟实现开始
- C#算法之关于大牛生小牛的问题
- C#实现的算24点游戏算法实例分析
- 经典排序算法之冒泡排序(Bubble sort)代码
- c语言实现的带通配符匹配算法
- 浅析STL中的常用算法
- 算法之排列算法与组合算法详解
- C++实现一维向量旋转算法
- Ruby实现的合并排序算法
- C#折半插入排序算法实现方法
- 基于C++实现的各种内部排序算法汇总
- C++线性时间的排序算法分析
- C++实现汉诺塔算法经典实例