Skip to content
md
给你一个长度为 n 的整数数组 nums 和一个正整数 k 。

一个数组的 能量值 定义为:

如果 所有 元素都是依次 连续(即 nums[i] + 1 = nums[i + 1],i < n)且 上升 的,那么能量值为 最大 的元素。
否则为 -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

有一个, 接近正确答案的作法:

使用双层for循环, 会超时;

ts

/*
 * @lc app=leetcode.cn id=3255 lang=typescript
 * @lcpr version=30204
 *
 * [3255] 长度为 K 的子数组的能量值 II
 */

function resultsArray(nums: number[], k: number): number[] {
    let len = nums.length - k + 1
    let res:number[] = [];
    for(let i=0;i<len;i++){
        let isValid = true;
        for(let j=0;j<k-1;j++){
            if (nums[i + j] + 1 !== nums[i + j + 1]) {
                isValid = false;
                break;
            }
        }
        if(isValid){
            res.push(nums[i+k-1])
        }else{
            res.push(-1)
        }
    }
    return res;
};

优化成单层循环:

如果满足 i=0 或者 nums[i]−nums[i−1]=1 时,此时 cnt=cnt+1,即在 nums[i−1] 末尾可以追加元素 nums[i] 仍然满足连续上升;

nums[i]−nums[i−1] !== 1 时,此时 cnt 重新置为 1,即在 nums[i−1] 末尾无法追加 nums[i];

使用一个cnt的方式, 统一

ts
function resultsArray(nums: number[], k: number): number[] {
    const n = nums.length;
    let cnt = 0;
    const res = new Array(n-k+1).fill(-1);

    for(let i=0;i<n;i++){
        if( i===0 || nums[i] - nums[i-1] !== 1 ){
            cnt = 1;
        }else {
            cnt++;
        }
        if(cnt >= k){
            res[i-k+1] = nums[i]
        }
    }
    return res
};
本站访客数 人次 本站总访问量