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;
}