LeetCode-in-Java

3255. Find the Power of K-Size Subarrays II

Medium

You are given an array of integers nums of length n and a positive integer k.

The power of an array is defined as:

• Its maximum element if all of its elements are consecutive and sorted in ascending order.
• -1 otherwise.

You need to find the power of all subarrays of nums of size k.

Return an integer array results of size n - k + 1, where results[i] is the power of nums[i..(i + k - 1)].

Example 1:

Input: nums = [1,2,3,4,3,2,5], k = 3

Output: [3,4,-1,-1,-1]

Explanation:

There are 5 subarrays of nums of size 3:

• [1, 2, 3] with the maximum element 3.
• [2, 3, 4] with the maximum element 4.
• [3, 4, 3] whose elements are not consecutive.
• [4, 3, 2] whose elements are not sorted.
• [3, 2, 5] whose elements are not consecutive.

Example 2:

Input: nums = [2,2,2,2,2], k = 4

Output: [-1,-1]

Example 3:

Input: nums = [3,2,3,2,3,2], k = 2

Output: [-1,3,-1,3,-1]

Constraints:

• 1 <= n == nums.length <= 105
• 1 <= nums[i] <= 106
• 1 <= k <= n

Solution

public class Solution {
    public int[] resultsArray(int[] nums, int k) {
        if (k == 1) {
            return nums;
        }
        int start = 0;
        int n = nums.length;
        int[] output = new int[n - k + 1];
        for (int i = 1; i < n; i++) {
            if (nums[i] != nums[i - 1] + 1) {
                start = i;
            }
            int index = i - k + 1;
            if (index >= 0) {
                if (start > index) {
                    output[index] = -1;
                } else {
                    output[index] = nums[i];
                }
            }
        }
        return output;
    }
}

This site is open source. Improve this page.