题目
话说大诗人李白,一生好饮。幸好他从不开车。
一天,他提着酒壶,从家里出来,酒壶中有酒两斗。他边走边唱:
无事街上走,提壶去打酒。
逢店加一倍,遇花喝一斗。
这一路上,他一共遇到店 5 次,遇到花 10 次,已知最后一次遇到的是花,他正好把酒喝光了。请你计算李白遇到店和花的次序,有多少种可能的方案。
解析
这个问题用的是次数,可以遍历每一个可能。不能使用全排列将所有可能列出,太耗时。我们自己不用考虑酒店和花的次序,使用递归。
代码
#include<iostream>
using namespace std;
int ans = 0;
void function(int dian,int hua,int jiu){
if(dian>0)
function(dian-1,hua,jiu*2);
if(hua>0)
function(dian,hua-1,jiu-1);
if(dian == 0&&hua == 0&&jiu == 1)
ans++;
}
int main(){
function(5,9,2);
printf("%d",ans);
return 0;
}
变形
题目只要求得次数,如果还要输出每次符合条件的排列,如何求解?
很简单,只要对上式稍稍变形,添加一个数组,将遇到的是店还是花记录在数组上,满足最终条件即输出并加一。
代码:
#include<iostream>
using namespace std;
int ans = 0;
int arr[14];
void function(int dian,int hua,int jiu,int arr[],int i){
if(dian>0){
arr[i] = 1;
function(dian-1,hua,jiu*2,arr,i+1);
}
if(hua>0){
arr[i] = 0;
function(dian,hua-1,jiu-1,arr,i+1);
}
if(dian == 0&&hua == 0&&jiu == 1){
for(int j=0;j<14;j++){
printf("%d ",arr[j]);
}
printf("\n");
ans++;
}
}
int main(){
function(5,9,2,arr,0);
printf("%d",ans);
return 0;
}