【思特奇杯·云上蓝桥-算法集训营】第2周 第2题 李白打酒

第二题 李白打酒
问题描述

话说大诗人李白，一生好饮。幸好他从不开车。

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

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

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

这一路上，他一共遇到店5次，遇到花10次，已知最后一次遇到的是花，他正好把酒喝光了。

请你计算李白遇到店和花的次序，可以把遇店记为a，遇花记为b。则：babaabbabbabbbb 就是合理的次序。像这样的答案一共有多少呢？请你计算出所有可能方案的个数（包含题目给出的）。

解题思路：

• 遇到店，店的数量 - 1，同时酒的数量翻倍
  遇到花，花的数量 - 1，同时酒的数量 - 1
• 最后一次是遇到花
  则说明在第十四次结束时酒的数量和花的数量都剩余1

import itertools

res = 0
a = itertools.permutations(['a'] * 5 + ['b'] * 9)
for i in a:
    wine = 2
    for j in range(0, len(i)):
        if i[j] == 'a':
            wine *= 2
        elif i[j] == 'b':
            wine -= 1
        elif j == len(i) and wine == 1:
            res += 1

print(res)