李白打酒加强版(dp)

话说大诗人李白,一生好饮。幸好他从不开车。一天,他提着酒壶,从家里出来,酒壶中有酒 2斗。他边走边唱:

无事街上走,提壶去打酒。

逢店加一倍,遇花喝一斗。

这一路上,他一共遇到店 N次,遇到花 M次。已知最后一次遇到的是花,他正好把酒喝光了。请你计算李白这一路遇到店和花的顺序,有多少种不同的可能?

注意:壶里没酒 (0斗) 时遇店是合法的,加倍后还是没酒;但是没酒时遇花是不合法的。

输入格式

第一行包含两个整数 N和 M。

输出格式

输出一个整数表示答案。由于答案可能很大,输出模 1000000007的结果。

数据范围

对于 40%的评测用例:1≤N,M≤10。
对于 100% 的评测用例:1≤N,M≤100。

输入样例:

5 10

输出样例:

14

样例解释

如果我们用 0代表遇到花,1 代表遇到店,14种顺序如下:

010101101000000
010110010010000
011000110010000
100010110010000
011001000110000
100011000110000
100100010110000
010110100000100
011001001000100
100011001000100
100100011000100
011010000010100
100100100010100
101000001010100

 法一:

四维数组,i,j,k,c
时间复杂度:n*m*k*c=100*100*k*2;酒最多有102斗,约为2*1e6

李白打酒加强版(dp)_第1张图片

 ac代码

#include 
#include 
#include 

using namespace std;
int n,m;
int f[110][110][110][2];
const int M = 1000000007;

int main()
{
    cin>>n>>m;
    f[0][0][2][1]=1;
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=m;j++)
        {
            for(int k=0;k<=101;k++)
            {
                for(int c=0;c<2;c++)
                {
                    if(i > 0 && c == 1 && k % 2 == 0)
                    {
                        f[i][j][k][c] = (f[i - 1][j][k /2][0] + f[i - 1][j][k /2][1]) % M;
                    }
                    
                    if(j > 0  && c == 0 && k >= 0)
                    {
                        f[i][j][k][c] = (f[i][j - 1][k+1][0] + f[i][j - 1][k +1][1]) % M;
                    }
                }
            }
        }
    }
    
    cout<

 法二

三维
同上,减少了关于c的循环

注意:
此时答案为f[n][m-1][1],即还剩一次花未经过,还剩一斗酒

 

#include 
#include 
#include 

using namespace std;

int n,m;
int f[110][110][110];
const int N = 1000000007 ;
int main()
{
    cin>>n>>m;
    
    f[0][0][2]=1;
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=m;j++)
        {
            for(int k=0;k<=102;k++)
            {
                if(i>0&&k%2==0)
                f[i][j][k]=(f[i][j][k]+f[i-1][j][k/2])%N;
                if(j>0)
                f[i][j][k]=(f[i][j][k]+f[i][j-1][k+1])%N;
            }
        }
    }
    
    cout<

你可能感兴趣的:(算法,蓝桥杯,c++,算法,动态规划)