算法013:水果成篮

时间:2024-07-09 07:00:36

水果成篮. - 备战技术面试?力扣提供海量技术面试资源,帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/fruit-into-baskets/

这道题题目很长,仔细阅读过后,我们其实可以简化成:

找出一个最长的子数组的长度,子数组中不超过两种类型的水果。

相对于算法009来说,有很多相似的地方,不过不同的点在于,这个题还多了一个限制条件:子数组中不超过两种类型的水果。这是什么意思呢?

我们自己模拟出一个hash表后,还需要用一个参数kinds来记录当前不同类型的水果,当hash表中的值从0变成1的时候,kinds值+1,反之则-1。

右指针向右移动,移动到数字时,让hash表的值+1,如果这个数是第一次出现,则kinds的值+1

kinds的值大于2了,说明滑动窗口不符合条件了,此时左指针向右移动,并且移动到的数字让hash表的值-1,如果这个hash表的值为0了,说明这个数字不存在于滑动窗口里面了,此时让kinds-1.

最后记录下最长的滑动窗口的数字。

代码:

class Solution {
    public int totalFruit(int[] fruits) {
        int n = fruits.length;
        int[] hash = new int[n+1];
        int ret = 0;
        for(int left = 0, right = 0, kinds = 0; right < n; right++){
            int in = fruits[right];
            if(hash[in] == 0){
                kinds++;
            }
            hash[in]++;
            while(kinds > 2){
                int out = fruits[left];
                hash[out]--;
                if(hash[out] == 0){
                    kinds--;
                }
                left++;
            }
            ret = Math.max(ret, right - left + 1);
        }
        return ret;
    }
}