点击跳转力扣904
你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits
表示,其中 fruits[i]
是第 i
棵树上的水果 种类 。
你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
- 你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
- 你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次,你将会向右移动到下一棵树,并继续采摘。
- 一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
给你一个整数数组 fruits
,返回你可以收集的水果的 最大 数目。
示例 1:
输入:fruits = [1,2,1] 输出:3 解释:可以采摘全部 3 棵树。
示例 2:
输入:fruits = [0,1,2,2] 输出:3 解释:可以采摘 [1,2,2] 这三棵树。 如果从第一棵树开始采摘,则只能采摘 [0,1] 这两棵树。
示例 3:
输入:fruits = [1,2,3,2,2] 输出:4 解释:可以采摘 [2,3,2,2] 这四棵树。 如果从第一棵树开始采摘,则只能采摘 [1,2] 这两棵树。
示例 4:
输入:fruits = [3,3,3,1,2,1,1,2,3,3,4] 输出:5 解释:可以采摘 [1,2,1,1,2] 这五棵树。
提示:
1 <= fruits.length <= 105
0 <= fruits[i] < fruits.length
思路:
该题的意思是找出数组中两种水果所包含的最大数量,我们可以用一个哈希表来存储水果种类跟水果数量,当水果种类大于2时,移除一些'水果'
class Solution {
public:
int totalFruit(vector<int> &fruits) {
// 窗口长度、起始指针、当前果树种类计数
int wl = 0, i = 0, cnt = 0, len = fruits.size();
// 果篮
unordered_map<int, int> basket;//哈希表
// 终止指针遍历数组,
for (int j = 0; j < len; ++j) {
// 遇到果篮里没有的种类,窗口数量+1,水果总数+1
if (basket[fruits[j]] == 0)cnt++;
basket[fruits[j]]++;
while (cnt > 2) {
basket[fruits[i]]--;
// 果篮数大于2,则向右移动初始指针,缩小滑动窗口,更新水果总数,保证种类为2的情况下,窗口最大化.
if (basket[fruits[i]] == 0)cnt--;
i++;
}
wl = max(wl, j - i + 1);
}
return wl;
}
};
总结:
unordered_map函数,我简单一点理解就是哈希表,详细可以参考https://blog.csdn.net/hk2291976/article/details/51037095
补充:
这里补充一个知识点----滑动窗口:就是不断的调节子序列的起始位置和终止位置,从而得出结果.
实现滑动窗口的框架是定死的,用left,right
来刻画窗口,窗口区间为[left, right)
,注意为半开半闭。框架如下:
void slidwindow(vector<int> nums)
{
int left = 0, right = 0;
while(right < nums.size())
{
...//扩大右边界并更新窗口状态
right++;
while(需要收缩)//窗口到达什么状态需要收缩
{
...//缩小左边界并更新窗口状态
left++;
}
}
}
//以上框架来自力扣题解:凯巴尔我的超人
以力扣209为例:209. 长度最小的子数组 - 力扣(LeetCode)
给定一个含有 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 = 4, nums = [1,4,4] 输出:1
示例 3:
输入:target = 11, nums = [1,1,1,1,1,1,1,1] 输出:0
思路:
在本题中实现滑动窗口,主要确定如下三点:
- 窗口内是什么?
- 如何移动窗口的起始位置?
- 如何移动窗口的结束位置?
窗口就是 满足其和 ≥ s 的长度最小的连续子数组。
窗口的起始位置如何移动:如果当前窗口的值大于s了,窗口就要向前移动了(也就是该缩小了)。
窗口的结束位置如何移动:窗口的结束位置就是遍历数组的指针,也就是for循环里的索引。
解题的关键在于 窗口的起始位置如何移动,如图所示:
class Solution {
public:
int minSubArrayLen(int s, vector<int>& nums) {
int result = INT32_MAX;
int sum = 0; // 滑动窗口数值之和
int i = 0; // 滑动窗口起始位置
int subLength = 0; // 滑动窗口的长度
for (int j = 0; j < nums.size(); j++) {
sum += nums[j];
// 注意这里使用while,每次更新 i(起始位置),并不断比较子序列是否符合条件
while (sum >= s) {
subLength = (j - i + 1); // 取子序列的长度
result = result < subLength ? result : subLength;
sum -= nums[i++]; // 这里体现出滑动窗口的精髓之处,不断变更i(子序列的起始位置)
}
}
// 如果result没有被赋值的话,就返回0,说明没有符合条件的子序列
return result == INT32_MAX ? 0 : result;
}
};
以上解题思路来自<<代码随想录>>,有相关疑惑,可以去b站看up的讲解视频.下面讲一下我自己的思路,我个人认为更好理解这个代码思想.
我的思路是先算出数组的前缀和,当前缀和大于目标数时,减去数组左边的数,一直减到前缀和小于目标数,更新左指针,一旦前缀和小于目标数,则加上左指针所在的数,重复此操作,不断更新ans,直到该区间内为目标数的最小序列.文章来源:https://www.toymoban.com/news/detail-826930.html
class Solution {
public:
int minSubArrayLen(int target, vector<int>& nums) {
int n=nums.size();
int s=0,left=0,ans=n+1;
for(int i=0;i<n;i++){
s=s+nums[i];
if(s>=target){
while(left<=i){
s=s-nums[left];
if(s>=target){
left++;
}else{
s=s+nums[left];
ans=min(ans,i-left+1);
break;
}
}
}
}
if(ans!=n+1) return ans;
else return 0;
}
};
总结
滑动窗口的重要性是确定起始指针跟终止指针的位置,用一个for循环解决两个for循环的问题(类似于双指针)文章来源地址https://www.toymoban.com/news/detail-826930.html
到了这里,关于力扣904 水果成篮--滑动窗口+哈希表的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!