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 <= 105
  • 1 <= nums[i] <= 106
  • 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

相关推荐
ShiinaMashirol3 小时前
代码随想录打卡|Day27(合并区间、单调递增的数字、监控二叉树)
java·算法
wuqingshun3141595 小时前
蓝桥杯 5. 交换瓶子
数据结构·c++·算法·职场和发展·蓝桥杯
Demons_kirit5 小时前
Leetcode 2845 题解
算法·leetcode·职场和发展
adam_life6 小时前
http://noi.openjudge.cn/——2.5基本算法之搜索——200:Solitaire
算法·宽搜·布局唯一码
我想进大厂7 小时前
图论---朴素Prim(稠密图)
数据结构·c++·算法·图论
我想进大厂7 小时前
图论---Bellman-Ford算法
数据结构·c++·算法·图论
AIGC大时代7 小时前
高效使用DeepSeek对“情境+ 对象 +问题“型课题进行开题!
数据库·人工智能·算法·aigc·智能写作·deepseek
lkbhua莱克瓦247 小时前
用C语言实现——一个中缀表达式的计算器。支持用户输入和动画演示过程。
c语言·开发语言·数据结构·链表·学习方法·交友·计算器
CODE_RabbitV7 小时前
【深度强化学习 DRL 快速实践】近端策略优化 (PPO)
算法
Wendy_robot8 小时前
【滑动窗口+哈希表/数组记录】Leetcode 438. 找到字符串中所有字母异位词
c++·算法·leetcode