leetcode 3254 长度为 K 的子数组的能量值 I 中等

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

分析:由于数组的长度只有500,因此我们可以枚举当前下标向前k位是否连续且递增。

cpp 复制代码
/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* resultsArray(int* nums, int numsSize, int k, int* returnSize) {
    *returnSize=numsSize-k+1;
    int *result=(int*)malloc(sizeof(int)*numsSize-k+1);

    for(int i=k-1;i<numsSize;++i)
    {
        int f=1;
        for(int j=i,t=k-1;t>0;t--,j--)
        {
            if(nums[j]!=nums[j-1]+1)
            {
                f=-1;break;
            }
        }
        if(f==-1)result[i+1-k]=-1;
        else result[i+1-k]=nums[i];
    }

    return result;
}

当然,如果数据范围很大,枚举就会超时。实际上不需要每个点都枚举。

根据题意可知,由于子数组如果满足连续上升,此时相邻元素的差值一定为 1,此时我们在遍历数组的同时,用一个计数器 cnt 统计以当前索引 i 为结尾时连续上升的元素个数,初始化 cnt=0,此时:

如果满足 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];

在计算的同时,此时如果以 nums[i] 为结尾的连续上升的元素数组如果大于等于 k,则当前一定存在长度为 k 且以 nums[i] 为结尾的连续上升的子数组,此时能量值即为 nums[i];如果不满足,则无法构成长度为 k 且连续上升的子数组,则当前能量值为 −1,返回最终统计的能量值即可。

cpp 复制代码
/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* resultsArray(int* nums, int numsSize, int k, int* returnSize) {
    *returnSize=numsSize-k+1;
    int *result=(int*)malloc(sizeof(int)*numsSize-k+1);
    for(int i=0;i<*returnSize;++i)
        result[i]=-1;
    int cnt=0;
    for(int i=0;i<numsSize;++i)
    {
        if(i==0||nums[i]-nums[i-1]==1)cnt=cnt+1;
        else cnt=1;
        if(cnt>=k)result[i-k+1]=nums[i];
    }

    return result;
}
相关推荐
付宇轩2 小时前
leetcode 1470.重新排列数组
数据结构·算法·leetcode
2401_858286114 小时前
L6.【LeetCode笔记】合并两个有序链表
c语言·开发语言·笔记·leetcode·链表
__AtYou__4 小时前
Golang | Leetcode Golang题解之第541题反转字符串II
leetcode·golang·题解
OT.Ter4 小时前
【力扣打卡系列】单调栈
算法·leetcode·职场和发展·go·单调栈
益达爱喝芬达5 小时前
力扣11.4
数据结构·leetcode
Ddddddd_1586 小时前
C++ | Leetcode C++题解之第541题反转字符串II
c++·leetcode·题解
OT.Ter7 小时前
【力扣打卡系列】移动零(双指针)
算法·leetcode·职场和发展·go
_GR15 小时前
每日OJ题_牛客_最长公共子序列_DP_C++_Java
java·开发语言·数据结构·c++·算法·leetcode
ZShiJ15 小时前
【题解】—— LeetCode一周小结44
算法·leetcode·职场和发展