DP(比基础稍难练习?)

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;
}





  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值