DP(比基础稍难练习?)

http://acm.sdut.edu.cn/sdutoj/showproblem.php?pid=3565&cid=1761

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

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

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

使用记忆化搜索的方法。

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

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

//注意dp维数较多时,开的dp的大小要和用的时候对应
//因为开的dp维数的对应不对,屡wa不止
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
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<




你可能感兴趣的:(动态规划(dp))