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

【LetMeFly】3254.长度为 K 的子数组的能量值 I：等II再用O(n)的方法，先暴力模拟了再说

力扣题目链接：https://leetcode.cn/problems/find-the-power-of-k-size-subarrays-i/

给你一个长度为 n 的整数数组 nums 和一个正整数 k 。

一个数组的 能量值 定义为：

• 如果 所有 元素都是依次 连续 且 上升 的，那么能量值为 最大 的元素。
• 否则为 -1 。

你需要求出 nums 中所有长度为 k 的 子数组 的能量值。

请你返回一个长度为 n - k + 1 的整数数组 results ，其中 results[i] 是子数组 nums[i..(i + k - 1)] 的能量值。

 

示例 1：

输入：nums = [1,2,3,4,3,2,5], k = 3

输出：[3,4,-1,-1,-1]

解释：

nums 中总共有 5 个长度为 3 的子数组：

• [1, 2, 3] 中最大元素为 3 。
• [2, 3, 4] 中最大元素为 4 。
• [3, 4, 3] 中元素 不是 连续的。
• [4, 3, 2] 中元素 不是 上升的。
• [3, 2, 5] 中元素 不是 连续的。

示例 2：

输入：nums = [2,2,2,2,2], k = 4

输出：[-1,-1]

示例 3：

输入：nums = [3,2,3,2,3,2], k = 2

输出：[-1,3,-1,3,-1]

 

提示：

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

解题方法：模拟

用$i$从$0$到$n - k$枚举区间的起点，用$j$从$i+1$到$i+k-1$枚举区间中第一个元素之外的其他元素。

如果存在$j$使得$nums[j] != nums[j - 1] + 1$，就令$ans[i] = -1$；否则，令$ans[i] = nums[i + k - 1]$。

• 时间复杂度$O(nk)$
• 空间复杂度$O(1)$，力扣返回值不计入算法空间复杂度

$O(n)$时间复杂度的算法请看题解【LetMeFly】3255.长度为 K 的子数组的能量值 II：和官解思路不同的O(n)做法（附思考过程）。

AC代码

C++

class Solution {
public:
    vector<int> resultsArray(vector<int>& nums, int k) {
        vector<int> ans(nums.size() - k + 1);
        for (int i = 0; i + k <= nums.size(); i++) {
            for (int j = i + 1; j < i + k; j++) {
                if (nums[j] != nums[j - 1] + 1) {
                    ans[i] = -1;
                    break;
                }
            }
            if (ans[i] != -1) {
                ans[i] = nums[i + k - 1];
            }
        }
        return ans;
    }
};

Python

from typing import List

class Solution:
    def resultsArray(self, nums: List[int], k: int) -> List[int]:
        return [nums[i + k - 1] if all(nums[j] == nums[j - 1] + 1 for j in range(i + 1, i + k)) else -1 for i in range(len(nums) - k + 1)]

Java

class Solution {
    public int[] resultsArray(int[] nums, int k) {
        int[] ans = new int[nums.length - k + 1];
        for (int i = 0; i + k <= nums.length; i++) {
            for (int j = i + 1; j < i + k; j++) {
                if (nums[j] != nums[j - 1] + 1) {
                    ans[i] = -1;
                    break;
                }
            }
            ans[i] = ans[i] == -1 ? -1 : nums[i + k - 1];
        }
        return ans;
    }
}

Go

package main

func resultsArray(nums []int, k int) (ans []int) {
    ans = make([]int, len(nums) - k + 1)
    for i := 0; i + k <= len(nums); i++ {
        for j := i + 1; j < i + k; j++ {
            if nums[j] != nums[j - 1] + 1 {
                ans[i] = -1
                break
            }
        }
        if (ans[i] != -1) {
            ans[i] = nums[i + k - 1]
        }
    }
    return
}

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/143575677