[蓝桥杯2022初赛] 李白打酒加强版

题目描述

话说大诗人李白，一生好饮。幸好他从不开车。
一天，他提着酒壶，从家里出来，酒壶中有酒 2 斗。他边走边唱：
无事街上走，提壶去打酒。
逢店加一倍，遇花喝一斗。
这一路上，他一共遇到店 N 次，遇到花 M 次。
已知最后一次遇到的是花，他正好把酒喝光了。
请你计算李白这一路遇到店和花的顺序，有多少种不同的可能？
注意：壶里没酒( 0 斗) 时遇店是合法的，加倍后还是没酒；但是没酒时遇花是不合法的。
输入格式
输入包含多组测试数据。
对于每组测试数据，输入两个整数N 和M.
1 ≤ N, M ≤ 100。
输出格式
输出一个整数表示答案。由于答案可能很大，输出模1000000007 的结果。
输入样例

5 10

输出样例

14

样例说明
如果我们用 0 代表遇到花，1 代表遇到店，14 种顺序如下：
010101101000000
010110010010000
011000110010000
100010110010000
011001000110000
100011000110000
100100010110000
010110100000100
011001001000100
100011001000100
100100011000100
011010000010100
100100100010100
101000001010100

解题思路

状态设计：dp[i][j][k]的值表示遇到i家店，j朵花，酒壶中还剩k斗酒的可能情况数；
状态转移方程：当酒量k是奇数的时候,说明最后遇到的是花,因为遇到店会加一倍,即原酒量*2,故最后遇到店时酒量k一定是偶数.
故当酒量k是奇数的时候dp[i][j][k](j>1)=dp[i][j-1][k+1](j>1)

举例子,输入样例中n=5 m=10,输出为14
故dp[5][9][1]=14
酒量k=1是奇数,所以 所有可能情况里最后遇到的一定是花(可参考上面的样例说明)
故dp[5][9][1]=dp[5][8][2]=14

状态转移方程：当酒量k是偶数的时候,说明最后可能遇到的是花,也有可能遇到的是酒
最后遇到的是花的情况数为 dp[i][j-1][k+1](j>1);最后遇到的是店的情况数为dp[i-1][j][k/2](i>1&&k%2==0)
故当酒量k是偶数的时候dp[i][j][k]为二者的和
即dp[i][j][k]=dp[i-1][j][k/2](i>1&&k%2==0) + dp[i][j-1][k+1](j>1);
边界设计：除了dp[0][0][2]=1，其他元素全为0；
他一共遇到店 N 次，遇到花 M 次。已知最后一次遇到的是花， 他正好把酒喝光了；所以最后一次肯定遇到的是花，那么最后的结果便是dp[N][M-1][1];
并且酒壶中酒的容量不能超过M；

动态规划代码

#include <iostream>
using namespace std;
int a[105][105][105];
int main()
{
  // 请在此输入您的代码
  int i,j,k,n,m;
  cin>>n>>m;
  a[0][0][2]=1;
  for(i=0;i<=n;i++){
    for(j=0;j<=m;j++){
      for(k=1;k<=m;k++){//最后遇到花,故k最小为1,因为要把酒喝完,故k最大为m
      	if(k%2!=0){//当酒量k是奇数的时候,dp[i][j][k]=dp[i][j-1][k+1](j>1)
      	 	if(j>0)
          		a[i][j][k]=a[i][j-1][k+1];
      	}else{//当酒量k是偶数的时候,dp[i][j][k]为二者的和
	      	if(j>0)
	          a[i][j][k]=a[i][j-1][k+1];
	        if(i>0)
	          a[i][j][k]+=a[i-1][j][k/2];
      	} //输出模1000000007 的结果。
        a[i][j][k]= a[i][j][k]%1000000007;
      }
    }
  }
  cout<<a[n][m-1][1];
  return 0;
}

运行结果

作者自己的dfs代码

可以通过4个案例,加了剪枝同样是只能通过4个案例

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
//输出模 1000000007 的结果.
int sum=0,mm;
int jisuan(vector<int> &a){
    int jiu=2;
    for(int i=0;i<a.size();i++){
        if(a[i]==1)
            jiu*=2;
        else{
            if(jiu>0)
                jiu=jiu-1;
            else
                return -1;
        }
    }
    return jiu;
}
void dfs(vector<int> a,int n,int i){
    int index;
    vector<int> b;
    b=a;
    int jiu;
    if(n==0){
        // for(index=0;index<a.size();index++)
        //     cout<<a[index]<<" ";
        // cout<<endl;
        jiu=jisuan(a);
        if(jiu==1){
            if(sum<1000000007)
                sum++;
            else
                sum=(sum+1)%1000000007;
        }
    }else{
        for(index=i+1;index<a.size();index++){
            b[index]=1;
            if(jisuan(b)<=mm+1)
                dfs(b,n-1,index);
            b=a;
        }
    }
}
int main(int argc, char** argv) {
    int n,m;
    int i;
    cin>>n>>m;
    mm=m;
    vector<int> a(n+m-1),b;
    for(i=0;i<n+m-1;i++){
        a[i]=0;
    }
    b=a;
    for(i=0;i<n+m-1;i++){
        b[i]=1;
        dfs(b,n-1,i);
        b=a;
    }
    cout<<sum;
    return 0;
}