Leetcode 3255. Find the Power of K-Size Subarrays II

Espresso Macchiato2024-08-19 21:36

[Leetcode 3255. Find the Power of K-Size Subarrays II](#Leetcode 3255. Find the Power of K-Size Subarrays II)
- [1. 解题思路](#1. 解题思路)
- [2. 代码实现](#2. 代码实现)

题目链接：3255. Find the Power of K-Size Subarrays II

1. 解题思路

这一题是题目3254的进阶版，其实主要就是增加了算法复杂度。

整体上来说的话思路还是一个分段的思路，我们找到所有的满足条件的连续递增数列，显然他们之间不可能有交集。

然后，对于任何一个连续递增的数列，假设其长度为 m m m，如果 m < k m<k m<k，那显然无法构成长度不少于 k k k的数组，因此我们返回一个长度为 m m m的 − 1 -1 −1的数组即可，反之如果 m ≥ k m \geq k m≥k，那么其返回结果的前 k − 1 k-1 k−1个值就是 − 1 -1 −1，后面则就是原始数组当中的值了。

我们将所有数组拼凑在一起，然后去除掉前 k − 1 k-1 k−1个元素即为我们最终的解答了。

2. 代码实现

给出python代码实现如下：

python 复制代码

class Solution:
    def resultsArray(self, nums: List[int], k: int) -> List[int]:
        n = len(nums)
        i = 0
        ans = []
        while i < n:
            j = i
            while j < n-1 and nums[j+1] == nums[j]+1:
                j += 1
            m = j-i+1
            if m >= k:
                ans += [-1] * (k-1) + nums[i+k-1:j+1]
            else:
                ans += [-1] * m
            i = j+1
        return ans[k-1:]

提交代码评测得到：耗时1312ms，占用内存32.9MB。