LeetCode 3255.长度为 K 的子数组的能量值 II:和官解思路不同的O(n)做法(附思考过程)

【LetMeFly】3255.长度为 K 的子数组的能量值 II:和官解思路不同的O(n)做法(附思考过程)

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

给你一个长度为 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 <= 10^5^
  • 1 <= nums[i] <= 10^6^
  • 1 <= k <= n

解题方法:遍历(类似滑动窗口)

本题长度为 K 的子数组的能量值 II和上一题长度为 K 的子数组的能量值 I的唯一区别是数据量,本题数据量不支持 O ( n k ) O(nk) O(nk)时间复杂度的做法。

上一题中我们枚举每个长度为K的区间的左端点,之后遍历区间,若区间中存在"相邻不连续"的两个元素则能量值为-1,否则为区间的最后一个元素。每个区间都可能需要 O ( k ) O(k) O(k)的时间复杂度来完成结果的计算。

但是其实不难发现,区间(假设k很大区间很长)每次向右移动一个元素,区间中大量的元素是不变的,只有区间起始位置元素和结束位置元素发生了变化。

因此解决方案来了,我们使用一个额外的变量 n o t C o n t i n u e notContinue notContinue记录区间中不连续的"相邻两个元素"有多少对,这样在区间移动的时候只需要根据区间开头元素和末尾元素的变化情况就能在 O ( 1 ) O(1) O(1)的时间内算出新区间的"notContinue"。

如果一个区间的 n o t C o u t i n u e notCoutinue notCoutinue非零,则说明该区间中存在相邻不连续的元素对,区间"能量值"为 − 1 -1 −1;否则区间能量值为区间的最后一个元素(即最大元素)。

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

AC代码

C++
cpp 复制代码
class Solution {
public:
    vector<int> resultsArray(vector<int>& nums, int k) {
        vector<int> ans(nums.size() - k + 1);
        int notContinue = 0;
        for (int i = 1; i < k; i++) {
            notContinue += nums[i] != nums[i - 1] + 1;
        }
        ans[0] = notContinue ? -1 : nums[k - 1];
        for (int i = 1; i + k <= nums.size(); i++) {
            notContinue -= nums[i] != nums[i - 1] + 1;
            notContinue += nums[i + k - 1] != nums[i + k - 2] + 1;
            ans[i] = notContinue ? -1 : nums[i + k - 1];
        }
        return ans;
    }
};
Python
python 复制代码
from typing import List

class Solution:
    def resultsArray(self, nums: List[int], k: int) -> List[int]:
        ans = [0] * (len(nums) - k + 1)
        notCoutinue = sum(nums[i] != nums[i -  1] + 1 for i in range(1, k))
        ans[0] = -1 if notCoutinue else nums[k - 1]
        for i in range(1, len(nums) - k + 1):
            notCoutinue -= nums[i] != nums[i - 1] + 1
            notCoutinue += nums[i + k - 1] != nums[i + k - 2] + 1
            ans[i] = -1 if notCoutinue else nums[i + k - 1]
        return ans
Java
java 复制代码
class Solution {
    public int[] resultsArray(int[] nums, int k) {
        int[] ans = new int[nums.length - k + 1];
        int notCoutinue = 0;
        for (int i = 1; i < k; i++) {
            if (nums[i] != nums[i - 1] + 1) {
                notCoutinue++;
            }
        }
        ans[0] = notCoutinue > 0 ? -1 : nums[k - 1];
        for (int i = 1; i + k <= nums.length; i++) {
            if (nums[i] != nums[i - 1] + 1) {
                notCoutinue--;
            }
            if (nums[i + k - 1] != nums[i + k - 2] + 1) {
                notCoutinue++;
            }
            ans[i] = notCoutinue > 0 ? -1 : nums[i + k - 1];
        }
        return ans;
    }
}
Go
go 复制代码
package main

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

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

Tisfy:https://letmefly.blog.csdn.net/article/details/143591327

相关推荐
ö Constancy6 分钟前
设计LRU缓存
c++·算法·缓存
香菜大丸8 分钟前
leetcode 面试150之 156.LUR 缓存
算法
infiniteWei25 分钟前
【Lucene】搜索引擎和文档相关性评分 BM25 算法的工作原理
算法·搜索引擎·lucene
2402_871321951 小时前
MATLAB方程组
gpt·学习·线性代数·算法·matlab
Mongxin_Chan1 小时前
【Cpp】指针与引用
c++·算法
非自律懒癌患者1 小时前
Transformer中的Self-Attention机制如何自然地适应于目标检测任务
人工智能·算法·目标检测
SSL_lwz2 小时前
P11290 【MX-S6-T2】「KDOI-11」飞船
c++·学习·算法·动态规划
zhangpz_2 小时前
c ++零基础可视化——vector
c++·算法
理论最高的吻3 小时前
98. 验证二叉搜索树【 力扣(LeetCode) 】
数据结构·c++·算法·leetcode·职场和发展·二叉树·c
沈小农学编程3 小时前
【LeetCode面试150】——202快乐数
c++·python·算法·leetcode·面试·职场和发展