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();
}
}