鸡蛋掉落

/*
 * 思路：与普通的Java方法类似，这里用Java Stream进行动态规划的状态转移。
 */

import java.util.stream.IntStream;

public class EggDropStream {
    public int superEggDrop(int k, int n) {
        int[][] dp = new int[k + 1][n + 1];
        IntStream.rangeClosed(1, k).forEach(i -> IntStream.rangeClosed(1, n).forEach(j -> dp[i][j] = j));
        IntStream.rangeClosed(1, k).forEach(i -> IntStream.rangeClosed(1, n).forEach(j ->
            IntStream.rangeClosed(1, j).forEach(x -> dp[i][j] = Math.min(dp[i][j], 1 + Math.max(dp[i-1][x-1], dp[i][j-x])))));
        return dp[k][n];
    }

    public static void main(String[] args) {
        EggDropStream solution = new EggDropStream();
        System.out.println(solution.superEggDrop(1, 2)); // 输出：2
        System.out.println(solution.superEggDrop(2, 6)); // 输出：3
        System.out.println(solution.superEggDrop(3, 14)); // 输出：4
    }
}