Leet code 904 水果成篮

最新推荐文章于 2024-09-04 21:39:45 发布

快乐的小飞只因

最新推荐文章于 2024-09-04 21:39:45 发布

阅读量462

点赞数 12

文章标签：哈希算法算法

chen

本文链接：https://blog.csdn.net/shaoxiebug/article/details/136748499

版权

文章讲述了如何使用滑动窗口策略和unordered_map数据结构解决一个关于数组中最多能包含多少种不同种类水果的问题，通过维护一个hash表来记录每种水果的数量，当种类数超过2时，逐个移除最早出现的种类直到种类数减为2。

摘要由CSDN通过智能技术生成

解题思路：滑动窗口

创建hash表记录水果的种类和数量

两个指针left 和 right 从数组起始位置开始滑动，right在循环内++

hash[fruits[right]]++ 进窗口

然后判断 hash表中的种类是否大于2

一旦大于2就要想办法出窗口出窗口需要将hash表中种类重回2种这时候就要动用left指针

然后更新结果

代码如下

class Solution {
public:
    int totalFruit(vector<int>& fruits) 
    {
       unordered_map<int,int> hash;//前面存种类 ，后面存数量
      int ret=0;
      for(int left=0,right=0 ;right<fruits.size();right++)
      {
        hash[fruits[right]]++;//进窗口
        while(hash.size()>2)
        {
            hash[fruits[left]]--;
            if(hash[fruits[left]]==0)//出窗口
            {
                hash.erase(fruits[left]);
            }
            left++;
        }
         ret=max(ret,right-left+1);
      }
      return ret;
    }
};