石子游戏 IV

/*
 * Alice 和 Bob 两个人轮流玩一个游戏，Alice 先手。
 * 一开始，有 n 个石子堆在一起。每个人轮流操作，正在操作的玩家可以从石子堆里拿走 任意 非零 平方数 个石子。
 * 如果石子堆里没有石子了，则无法操作的玩家输掉游戏。
 * 给你正整数 n ，且已知两个人都采取最优策略。如果 Alice 会赢得比赛，那么返回 True ，否则返回 False 。
 */
import java.util.stream.IntStream;

public class StoneGameStream {
    // 使用 Java Stream API 来解决这个问题
    public boolean winnerSquareGame(int n) {
        boolean[] dp = new boolean[n + 1];
        IntStream.rangeClosed(1, n).forEach(i -> {
            IntStream.iterate(1, k -> k * k <= i, k -> k + 1)
                     .filter(k -> !dp[i - k * k])
                     .findFirst()
                     .ifPresent(k -> dp[i] = true);
        });
        return dp[n];
    }

    public static void main(String[] args) {
        StoneGameStream game = new StoneGameStream();
        System.out.println(game.winnerSquareGame(1));  // true
        System.out.println(game.winnerSquareGame(2));  // false
        System.out.println(game.winnerSquareGame(4));  // true
        System.out.println(game.winnerSquareGame(7));  // false
        System.out.println(game.winnerSquareGame(17)); // false
    }
}