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

相关推荐
jianbaigreat1 小时前
代码随想录打卡Day22、23、24、25
数据结构·算法
Ddddddd_1581 小时前
C++ | Leetcode C++题解之第560题和为K的子数组
c++·leetcode·题解
_OLi_1 小时前
力扣 LeetCode 206. 反转链表(Day2:链表)
算法·leetcode·链表
运维&陈同学1 小时前
【HAProxy05】企业级反向代理HAProxy调度算法之静态算法与动态算法
linux·运维·算法·nginx·云原生·负载均衡·lvs·haproxy
weixin_478689761 小时前
【贪心算法】——力扣763. 划分字母区间
算法·leetcode·贪心算法
sp_fyf_20241 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-11-03
人工智能·深度学习·神经网络·算法·机器学习·语言模型·数据挖掘
友大冰2 小时前
前端开发中的CSS框架:昔日辉煌与新兴潮流
前端·css·算法·开源·tensorflow
nuyoah♂2 小时前
DAY27|贪心算法Part01|LeetCode:455.分发饼干、376. 摆动序列、53. 最大子序和
算法·leetcode·贪心算法
十七算法实验室2 小时前
Matlab实现鼠群优化算法(ROS)求解路径规划问题
开发语言·算法·决策树·支持向量机·matlab·动态规划·启发式算法
GeekAlice3 小时前
算法笔记/USACO Guide GOLD金组Graphs并查集Disjoint Set Union
c++·经验分享·笔记·学习·算法