打地鼠

/*
 * 思路：
 * 1. 使用流式编程对数据进行过滤和映射。
 * 2. 使用动态规划 (DP) 数组记录最优状态。
 * 3. 最后使用流求解最大值。
 */

import java.util.stream.IntStream;

public class WhackAMoleStream {
    public int maxMoles(int[][] moles) {
        int[][][] dp = new int[1001][3][3];
        for (int[] mole : moles) {
            int t = mole[0], x = mole[1], y = mole[2];
            IntStream.range(0, 3).forEach(i -> IntStream.range(0, 3).forEach(j -> {
                int distance = Math.abs(x - i) + Math.abs(y - j);
                if (distance <= t) {
                    dp[t][x][y] = Math.max(dp[t][x][y], dp[t - distance][i][j] + 1);
                }
            }));
        }
        return IntStream.range(0, 1001).flatMap(t -> IntStream.range(0, 3).flatMap(i -> IntStream.range(0, 3).map(j -> dp[t][i][j]))).max().orElse(0);
    }
}