3254. 长度为 K 的子数组的能量值 I

import java.util.ArrayList;
import java.util.List;

/**
 * 3254. 长度为 K 的子数组的能量值 I
 * 中等
 * 相关标签
 * 相关企业
 * 提示
 * 给你一个长度为 n 的整数数组 nums 和一个正整数 k 。
 * <p>
 * 一个数组的 能量值 定义为：
 * <p>
 * 如果 所有 元素都是依次 连续 且 上升 的，那么能量值为 最大 的元素。
 * 否则为 -1 。
 * 你需要求出 nums 中所有长度为 k 的
 * 子数组
 * 的能量值。
 * <p>
 * 请你返回一个长度为 n - k + 1 的整数数组 results ，其中 results[i] 是子数组 nums[i..(i + k - 1)] 的能量值。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [1,2,3,4,3,2,5], k = 3
 * <p>
 * 输出：[3,4,-1,-1,-1]
 * <p>
 * 解释：
 * <p>
 * nums 中总共有 5 个长度为 3 的子数组：
 * <p>
 * [1, 2, 3] 中最大元素为 3 。
 * [2, 3, 4] 中最大元素为 4 。
 * [3, 4, 3] 中元素 不是 连续的。
 * [4, 3, 2] 中元素 不是 上升的。
 * [3, 2, 5] 中元素 不是 连续的。
 * 示例 2：
 * <p>
 * 输入：nums = [2,2,2,2,2], k = 4
 * <p>
 * 输出：[-1,-1]
 * <p>
 * 示例 3：
 * <p>
 * 输入：nums = [3,2,3,2,3,2], k = 2
 * <p>
 * 输出：[-1,3,-1,3,-1]
 * <p>
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= n == nums.length <= 500
 * 1 <= nums[i] <= 10^5
 * 1 <= k <= n
 */
public class LeetCode1106_3254 {
    public static void main(String[] args) {
        int[] nums = {1, 3, 4};
        int k = 2;
        System.out.print("[");
        for (int i : resultsArray(nums, k)) {
            System.out.print(i + ",");
        }
        System.out.print("]");
    }

    /**
     * 长度为 K 的子数组的能量值 I
     *
     * @param nums 数组
     * @param k    长度
     * @return 符合条件的数组
     */
    public static int[] resultsArray(int[] nums, int k) {
        List<Integer> resultList = new ArrayList<>();
        if (nums.length < k) {
            return new int[0]; // 如果数组长度小于k，直接返回空数组
        }

        // 滑动窗口初始化
        int windowStart = 0;
        for (int windowEnd = 0; windowEnd <= nums.length - k; windowEnd++) {
            if (windowEnd != windowStart) {
                // 每次移动窗口时，只需要比较新加入的元素与前一个元素
                if (nums[windowEnd + k - 1] <= nums[windowEnd + k - 2]) {
                    resultList.add(-1); // 如果不是严格上升，则添加-1
                    continue;
                }
            }

            // 检查当前窗口内的元素是否连续且上升
            boolean increasing = true;
            for (int i = windowStart + 1; i < windowStart + k; i++) {
                if (nums[i] != nums[i - 1] + 1) {
                    increasing = false;
                    break;
                }
            }
            
            // 添加窗口最后一个元素作为能量值
            if (increasing) {
                resultList.add(nums[windowStart + k - 1]);
            } else {
                resultList.add(-1);
            }
            windowStart++;
        }

        return resultList.stream().mapToInt(i -> i).toArray();
    }
}