蓝桥杯——李白饮酒及变体升级（递归）

题目

话说大诗人李白，一生好饮。幸好他从不开车。
一天，他提着酒壶，从家里出来，酒壶中有酒两斗。他边走边唱：

无事街上走，提壶去打酒。
逢店加一倍，遇花喝一斗。

这一路上，他一共遇到店 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;
}