李白打酒加强版(dfs暴力枚举+dp)

题目

话说大诗人李白，一生好饮。幸好他从不开车。
一天，他提着酒壶，从家里出来，酒壶中有酒 2斗。他边走边唱：
无事街上走，提壶去打酒。逢店加一倍，遇花喝一斗。
这一路上，他一共遇到店 N次，遇到花 M次。已知最后一次遇到的是花，他正好把酒喝光了。

请你计算李白这一路遇到店和花的顺序，有多少种不同的可能？
注意：壶里没酒 (0斗) 时遇店是合法的，加倍后还是没酒；但是没酒时遇花是不合法的。
输入格式

第一行包含两个整数 N和 M。
输出格式

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

数据范围

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

5 10

输出样例：

14

题解

1.暴力枚举(2^n) --40%

#include <iostream>
using namespace std;
typedef long long ll;
ll res = 0;
ll N, M;

void dfs(ll flower, ll wine, ll i, ll remain)  //遇到多少次花，遇到多少店，当前是第几步，还有多少酒
{ 
	if (flower > M||wine > N) return;
	if (flower == M - 1 && wine == N && i == N + M - 1 && remain == 1)  //走到倒数第二步还剩1斗酒，且最后一步一定是花
	{
		res++;
		return;
	}
	//下次遇花
	if(remain>0)
	{
		dfs(flower + 1, wine, i + 1, remain - 1);
	}
	//下次遇店
	dfs(flower, wine+1, i + 1, remain * 2);
}
int main()
{
	cin >> N>> M;
	dfs(0,0,0,2);
	cout << res %1000000007<< endl;
	return 0;
}

2.dp三重循环O(n^3) --100%

#include<iostream>
using namespace std;

const int mod = 1000000007;
const int N = 110, M = 110;
typedef long long LL;
int dp[N][N][N+M];
int n, m;
int main()
{
	cin >> n >> m;
	dp[0][0][2] = 1;
	for (int i = 0; i <= m; i++)  //花
	{
		for (int j =0; j <= n; j++) //店
		{
			for (int k = 0; k <= 200; k++) //余酒,因为到最后一步(n+m)必须喝完酒,所以酒的数量不会超过最大值200;
			{
				if (i) 
				{
                     dp[i][j][k] = dp[i-1][j][k + 1] % mod ;
				}
				//只有当前酒为偶数时，上次才有可能遇酒
				if (k % 2 == 0&&j)
				{
					dp[i][j][k] = (dp[i][j][k]+dp[i][j - 1][k / 2])%mod;
				}
			}
		}
	}
	//倒数第二步，必须只剩一斗酒，且还剩一次花
	cout << dp[m - 1][n][1] << endl;
	return 0;
}

原题链接

李白打酒加强版