高楼扔鸡蛋-另解

问题求解的是当拥有k个鸡蛋和N层楼可供测试时，所需的最少扔鸡蛋个数m。那么存在这样的的一种思路：在k不变的情况下，由于m的增加必然会增加测试的楼层数，那么我们可以通过（k,m）状态下所需的楼层数来进行求解。当所需测试楼层数=N时，我们便可以得到最少扔鸡蛋个数。

import numpy as np


def egg_break(N, k):
    """

    :param N: 总楼层数
    :param k: 鸡蛋个数
    :return: 最坏情况下扔鸡蛋个数
    """

    m = 0
    dp = np.zeros((k + 1, N + 1), dtype=np.int)

    while dp[k][m] < N:
        m += 1
        for i in range(1, k + 1):
            dp[i][m] = dp[i][m - 1] + dp[i - 1][m - 1] + 1

    return m


if __name__ == '__main__':
    print('最坏情况下所需扔鸡蛋数：{}'.format(egg_break(100, 2)))
    print('最坏情况下所需扔鸡蛋数：{}'.format(egg_break(100, 4)))
    print('最坏情况下所需扔鸡蛋数：{}'.format(egg_break(200, 2)))