Leetcode每天打卡第14天

题目

长度为 K 的子数组的能量值 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

思路

根据题意可知，我们需要找到长度为 k 且满足连续上升的子数组中的最大元素，此时直接枚举数组中所有长度为 k 的子数组，并判断该数组是否满足连续且上升即可。子数组满足连续上升，此时子数组中相邻元素的差值一定为 1，即满足 nums[j]−nums[j−1]=1。如果子数组满足连续上升，此时最大值即为子数组中的最后一个元素，能量值即为子数组的最后一个元素；如果不满足，则此时能量值为 −1，按照上述描述依次枚举并返回能量值即可。

python3代码

class Solution:
    def resultsArray(self, nums: List[int], k: int) -> List[int]:
        n = len(nums)
        ans = [-1] * (n - k + 1)
        for i in range(n - k + 1):
            valid = True
            for j in range(i + 1, i + k):
                if nums[j] - nums[j - 1] != 1:
                    valid = False
                    break
            if valid:
                ans[i] = nums[i + k - 1]
                
        return ans