《程序员的算法趣题》-(日)增井敏克 Python解题 -- (Q18)

《程序员的算法趣题》-(日)增井敏克 ， 书中为69 道数学谜题编写了解题程序， 编程语言为：Ruby，JavaScript，C语言。有兴趣的同学，可以购书阅读~

在此更新个人编写的Python版，仅供学习使用。（运行环境：Python3.6）

Q18 水果酥饼日

    日本每月的 22 日是水果酥饼日。因为看日历的时候， 22 日的上方刚好是 15 日，也就是“‘22’这个数字上面点缀着草莓”(如果将日语的 15 拆为 1 和 5 发音，则与日语“草莓”一词发音相同，而水果酥饼中最为著名的就是草莓酥饼。)

    切分酥饼的时候，要求切分后每一块上面的草莓个数都不相同。假设切分出来的 N 块酥饼上要各有“1~N 个（共 N(N ＋ 1)÷2 个草莓）”。但这里要追加一个条件，那就是“一定要使相邻的两块酥饼上的数字之和是平方数”。
    举个例子，假设 N ＝ 4 时采用如 图的切法。这时，虽然 1 ＋ 3 ＝4 得到的是平方数，但“1 和 4” “2和 3” “2 和 4”的部分都不满足条件。

                                                              

问题
      求可以使切法满足条件的最小的 N（N ＞ 1）。

import math

def check_cake_list(start_cake, use_list, sqrt_list):
    for index, number in enumerate(use_list):
        if (start_cake + number) in sqrt_list:
            res_cakes = use_list[:index] + use_list[index+1:]
            if len(res_cakes) == 0:
                if (number + 1) in sqrt_list:
                    return True
            if check_cake_list(number, res_cakes, sqrt_list):
                return True

    return False

N = 2
while True:
    cake_list = [i for i in range(2, N+1)]
    sqrt_numbers = [i*i for i in range(1, int(math.sqrt(N*2))+2)]

    if check_cake_list(1, cake_list, sqrt_numbers):
        break

    N += 1

print("最小的N为：%s" % N)

运行结果：

                 最小的N为：32