当前位置：首页 > news >正文

904. 水果成篮

news 2025/8/11 10:14:31

一、题目

在这里插入图片描述

二、思路

2.1 解题思路

2.2 代码尝试

#include <map>
class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        std::map<int, int> pla;
       
        
        int l=0,r=0;
        int cur=0;//统计当前收集的水果数目
        int ret=0;//统计采摘的最大数目
        //开始采集，先按当前的第一个采集，统计窗口内采集情况
        while(r<fruits.size()){
            //窗口内每次遍历一个值，就把他加入哈希表，并维护哈希表
            pla[fruits[r]]++;
            
            //如果哈希表的长度大于2，说明需要把l键删除
            if(pla.size()<=2){
                cur++;
                ret=max(cur,ret);
            }else{
                cur--;
                pla.erase(fruits[l]);
                l++;
                r++;
                ret=max(cur,ret);
            }
            
            //判断如果哈希表长度超出2或者篮子数目不够了

        }

        //如果当前窗口不满足，限制了右指针的发展，那就更新左指针，然后再在此基础上继续更新右指针滑动窗口 

        return ret;
    }
};

想用哈希表来维护滑动窗口中的数据

2.3 疑难问题

三、解法

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int n = fruits.size();
        unordered_map<int, int> cnt;

        int left = 0, ans = 0;
        for (int right = 0; right < n; ++right) {
            ++cnt[fruits[right]];
            while (cnt.size() > 2) {
                auto it = cnt.find(fruits[left]);
                --it->second;
                if (it->second == 0) {
                    cnt.erase(it);
                }
                ++left;
            }
            ans = max(ans, right - left + 1);
        }
        return ans;
    }
};

作者：力扣官方题解
链接：https://leetcode.cn/problems/fruit-into-baskets/solutions/1893352/shui-guo-cheng-lan-by-leetcode-solution-1uyu/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。