- [Leetcode 3255. Find the Power of K-Size Subarrays II](#Leetcode 3255. Find the Power of K-Size Subarrays II)
- [1. 解题思路](#1. 解题思路)
- [2. 代码实现](#2. 代码实现)
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。