力扣904 水果成篮--滑动窗口+哈希表

点击跳转力扣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,直到该区间内为目标数的最小序列.

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循环的问题(类似于双指针)