LeetCode 每日一题 长度为 K 的子数组的能量值
长度为 K 的子数组的能量值
给你一个长度为 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
题解
看到题目要求关于定长的子数组,俺首先想到的是使用滑动窗口
但是维护窗口,也就是判断窗口是否都连续且上升很难做到,所以考虑其他思路
使用一次遍历,记录连续的元素的个数
使用变量 n=0 来记录每次循环中连续的元素的个数
数组 arr 为返回值
int f = 0 为数组 arr 下标
使用 for i 循环遍历数组 nums
如果 i==0 || nums[ i ] = nums[ i-1 ] + 1
则 n++ 即从 nums[ i ] 到左边的连续且上升的元素的个数为 n
否则
n = 1 即从 nums[ i ] 到左边的连续且上升的元素的个数为 1
假如 n >= k,则说明此时的子数组所有元素都是连续的,能量值为 nums[ i ]
arr[f++] = nums[i]
否则 假如 i>=k-1 (这个判断保证有子数组的长度再对arr进行赋值),则说明此时子数组内有不连续且上升的,能量值为-1
arr[f++] = -1
返回 arr 即可
代码如下↓
/*** Note: The returned array must be malloced, assume caller calls free().*/
int* resultsArray(int* nums, int numsSize, int k, int* returnSize) {int* arr = (int*)malloc(sizeof(int)*(numsSize-k+1));*returnSize=numsSize-k+1;int n=0;int f=0;for(int i=0;i<numsSize;i++){if(i==0 || nums[i]==nums[i-1]+1){n+=1;}else{n=1;}if(n>=k){arr[f++]=nums[i];}else if(i>=k-1){arr[f++]=-1;}}return arr;
}