李白打酒加强版【dp】

话说大诗人李白，一生好饮。

幸好他从不开车。

一天，他提着酒壶，从家里出来，酒壶中有酒 2 斗。

他边走边唱：

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

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

这一路上，他一共遇到店 N 次，遇到花 M 次。

已知最后一次遇到的是花，他正好把酒喝光了。

请你计算李白这一路遇到店和花的顺序，有多少种不同的可能？

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

输入格式

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

输出格式

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

数据范围

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

输入样例：

5 10

输出样例：

14

样例解释

如果我们用 00 代表遇到花，11 代表遇到店，1414 种顺序如下：

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

题解：

设置状态转移dp[i][j][k][0/1]表示前i个中还剩j个旅馆，此时为还剩k斗酒，0表示当前位置经过花，酒减一斗，1表示经过旅馆，斗数增一倍，然后枚举j和k即可,四种状态转移：

dp[i][j][k][0]+=dp[i-1][j][k+1][0];

dp[i][j][k][0]+=dp[i-1][j][k+1][1];

if(k%2==0) dp[i][j][k][1]+=dp[i-1][j+1][k/2][0];

if(k%2==0) dp[i][j][k][1]+=dp[i-1][j+1][k/2][1];

AC代码：

#include <bits/stdc++.h>
#define sc(x) scanf("%d",&x)
#define sl(x) scanf("%lld",&x)
#define ll long long
#define pb push_back 
using namespace std;
const int Max=1e6+5;
const int mod=1e9+7;
ll dp[205][205][105][2];
int main(){
	int n,m;sc(n);sc(m);
	dp[0][n][2][0]=1;
	for(int i=1;i<=n+m;i++){
		for(int j=n;j>=0;j--){
			for(int k=100;k>=0;k--){
				dp[i][j][k][0]+=dp[i-1][j][k+1][0];
				dp[i][j][k][0]%=mod;
				dp[i][j][k][0]+=dp[i-1][j][k+1][1];
				dp[i][j][k][0]%=mod;
				if(k%2==0){
					dp[i][j][k][1]+=dp[i-1][j+1][k/2][0];dp[i][j][k][1]%=mod;
					dp[i][j][k][1]+=dp[i-1][j+1][k/2][1];dp[i][j][k][1]%=mod;
				}
			}
		}
	}
	cout<<dp[n+m][0][0][0]<<endl;
}