李白打酒加强版
题目描述
话说大诗人李白,一生好饮。幸好他从不开车。
一天,他提着酒壶,从家里出来,酒壶中有酒 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;
}