剑指 Offer II 008. 和大于等于 target 的最短子数组
给定一个含有 n 个正整数的数组和一个正整数 target 。
找出该数组中满足其和 ≥ target 的长度最小的 连续子数组 [numsl, numsl+1, …, numsr-1, numsr] ,并返回其长度。如果不存在符合条件的子数组,返回 0 。
原题链接: 【点此查看】
示例1:
输入:target = 7, nums = [2,3,1,2,4,3]
输出:2
解释:子数组 [4,3] 是该条件下的长度最小的子数组。
示例2:
输入:target = 11, nums = [1,1,1,1,1,1,1,1]
输出:0
分析
定义两个指针 start 和 end分别表示子数组(滑动)的开始位置和结束位置,维护变量 sum 存储子数组中的元素和(即从 nums[start] 到 nums[end]的元素和)。
初始状态下,start 和 end 都指向下标 0,sum的值为 0。
每一轮迭代,将 nums[end] 加到 sum,如果 sum≥s,则更新子数组的最小长度(此时子数组的长度是 end−start+1),然后将 nums[start] 从 sum中减去并将 start右移,直到 sum<s,在此过程中同样更新子数组的最小长度。在每一轮迭代的最后,将 end右移。
题解
class Solution {
public:
int minSubArrayLen(int target, vector<int>& nums) {
int n = nums.size();
if (n <= 0) return 0;
int start = 0, end = 0;
int sum = 0, ans = INT_MAX;
while (end < n)
{
sum += nums[end];
while (sum >= target)
{
ans = std::min(ans, end - start + 1);
sum -= nums[start];
start++;
}
end++;
}
return ans == INT_MAX ? 0 : ans;
}
};
图示:
剑指 Offer II 009. 乘积小于 K 的子数组
题目:
给定一个正整数数组 nums和整数 k ,请找出该数组内乘积小于 k 的连续的子数组的个数。
原题链接: 【点此查看】
示例:
输入: nums = [10,5,2,6], k = 100
输出: 8
解释: 8 个乘积小于 100 的子数组分别为: [10], [5], [2], [6], [10,5], [5,2], [2,6], [5,2,6]。
需要注意的是 [10,5,2] 并不是乘积小于100的子数组。
分析
和上面的题是一个类型,使用滑动窗口求解。
枚举子数组的右端点 j,并且左端点从 i=0 开始,用 sum 记录子数组 [i,j]的元素乘积。每枚举一个右端点j,如果当前子数组元素乘积 sum 大于等于 k,那么我们右移左端点 i 直到满足当前子数组元素乘积小于 k 或者 i>j,那么元素乘积小于 k 的子数组数目为 j−i+1。返回所有数目之和。
题解
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int n = nums.size();
if (n <= 0 || k <= 1) return 0; // 1
int start = 0, end = 0; // 2
int sum = 1, ans = 0; // 3
while (end < n)
{
sum *= nums[end++]; // 4
while (sum >= k)
{
sum /= nums[start++]; // 5
}
ans += end - start; // 6
}
return ans;
}
};
- 边界条件判断。
- 定义窗口的两个左右指针
- 定义中间辅助变量用来记录乘积,再定义返回值。
- 外层循环先让end走,记录sum的变化(这里让end后置++了)。
- 循环判断,如果乘积大于等于k了,就让窗口左边往后移。
- 退出内层循环时,乘积小于k的子数组个数为 end - start + 1,但前面对end提前加一操作,所以是 end - start。