李白打酒——C++

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

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

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

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

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

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

 #include<iostream>
#include<string>
#include<set>
using namespace std;
set<string> result;
string road="               ";
string key = "ab";
int a = 0, b = 0;
int judge(string a) {
    int w = 2;
    for (int i = 0; i < a.length(); i++) {
        if (a[i] == 'a')
            w *= 2;
        else
            w--;
        if (w < 0)
            return 0;
    }
    if (w == 0)
        return 1;
    return 0;
}
void search(char ch,int a,int b, int k) {
    if (a < 5) {
            road [k]= 'a';
            search(key[0], a + 1, b, k + 1);
    }
    if (b < 10) {
            road[k]= 'b';
            search(key[1], a, b + 1, k + 1);
    }
    if (judge(road))
        result.insert(road);
}
int main() {
    search('a',0,0,0);
    for (auto it = result.begin(); it != result.end(); it++)
        cout << *it << endl;
    return 0;
}