Leetcode每日刷题之904.水果成篮

ZZZ_O^O2024-08-23 20:41

1.题目解析

本题的题目要求较长，不过理解起来较为简单，就是在给定数组内找出最长子数组，并且该最长子数组只能有两种数字，最后返回该符合条件的最长子数组的长度即可

题目来源： 904.水果成篮

2.算法原理

本题的核心是找出符合条件的最长子数组，首先想到的一定是暴力枚举所有符合条件的子数组最后找出最长的即可

在经过优化的算法就是"滑动窗口"算法，

1.即使用两个指针，让right指针不断向后遍历而left指针不动，在left-right这一段窗口内加入数据称为入窗口

2.当窗口内数字的种类超过两种则开始出窗口，也就是left指针移动而right指针不动，直到窗口中数字种类为2种，这时更新子数组长度

3.这里使用数组hash模仿哈希表，因为由题目可知数组中数字范围小于10^5，所以创建一个hash[100001]即可，然后使用kinds变量统计窗口内数字种类

4.hash[100001]存储每个数字出现的次数，当出窗口时某个数字出现次数为0则将种类kinds--即可，最后更新长度直到找出最长符合子数组的长度即可

3.代码展示

cpp 复制代码

class Solution {
public:
    int totalFruit(vector<int>& fruits) 
    {
        int n = fruits.size();
        int hash[100001] = { 0 };
        int len = 0;
        for(int left = 0,right = 0,kinds = 0;right < n;right++)
        {
            if(hash[fruits[right]] == 0)
            {
                kinds++;
            }
            hash[fruits[right]]++;
            while(kinds > 2)
            {
                hash[fruits[left]]--;
                if(hash[fruits[left]] == 0)
                {
                    kinds--;
                }
                left++;
            }
            len = max(len,right - left + 1);
        }
        return len;
    }
};