Java、python实现啊哈算法 —— chapter3 火柴棍等式

逻辑书上写的已经很清楚了，但是书上的代码里有一点错误，最大能取到的数字是‘11111’ 而不是‘1111’，好了，下面上代码：

**python：**

def fun(x): #初始化一个方法用来查看某个搭建某个数字需要多少根火柴棍
    num = 0 #初始化一个变量存储数字x需要的总的火柴根数
    f = [6, 2, 5, 5, 4, 5, 6, 3, 7, 6]
    if x // 10 != 0: #如果这个数是两位数以上，需要从个位数开始逐个计算每个单独的数字需要的火柴个数
        while x // 10 != 0: #当x是两位数以上时
            digit = x % 10 #取x的最后一个数字
            num += f[digit] #将搭建这个数字需要的火柴数加到需要的总根数里
            x = x // 10 #将x中除了个位数字之外的数字重新赋值给x
        num += fun(x) #退出while循环时，x是原数字中最高位的数，将这个数字的火柴根数加到总根数里面
    else:
        num += f[x] #这个数只有一位数字
    return num

if __name__ == '__main__':
    m = int(input('输入火柴根数（不超过24）：'))
    res = 0  # 初始化一个变量存储符合条件的火柴棍等式的个数
    for a in range(11112):
        for b in range(11112):
            c = a + b
            if fun(a) + fun(b) + fun(c) == m - 4:
                res += 1
                print(str(a) + ' + ' + str(b) + ' = ' + str(c))
            else:
                continue
    print('满足条件的等式有' + str(res) + '个')