最大休假天数

/*
题目思路:
1. 使用动态规划结合Java Stream API来解决最大休假天数的问题。
2. 定义一个二维数组dp，其中dp[i][j]表示在第i周到城市j所能获得的最大休假天数。
3. 初始化第一周的休假天数。
4. 从第二周开始，更新dp数组，根据上一周的结果计算当前周的最大休假天数。
5. 最终，从dp数组的最后一行中找到最大的休假天数。
*/
 
import java.util.Arrays;
 
public class MaxVacationDaysStream {
    public int maxVacationDays(int[][] flights, int[][] days) {
        int N = flights.length; // 城市数量
        int K = days[0].length; // 周数
        int[][] dp = new int[K][N];
        // 初始化dp数组
        for (int i = 0; i < N; i++) {
            dp[0][i] = (flights[0][i] == 1 || i == 0) ? days[i][0] : -1;
        }
        // 计算从第二周到第K周的最大休假天数
        for (int week = 1; week < K; week++) {
            int[] temp = new int[N];
            Arrays.fill(temp, -1);
            for (int city = 0; city < N; city++) {
                int finalCity = city;
                temp[city] = Arrays.stream(dp[week - 1]).flatMap(prevDays -> {
                    int prevCity = Arrays.asList(dp[week - 1]).indexOf(prevDays);
                    return (prevDays != -1 && (flights[prevCity][finalCity] == 1 || finalCity == prevCity))
                        ? Arrays.stream(new int[]{prevDays + days[finalCity][week]})
                        : Arrays.stream(new int[]{-1});
                }).max().orElse(-1);
            }
            dp[week] = temp;
        }
        // 找到最后一周的最大休假天数
        return Arrays.stream(dp[K - 1]).max().orElse(0);
    }
}