水果成籃. - 備戰技術面試?力扣提供海量技術面試資源,幫助你高效提升編程技能,輕松拿下世界 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;}
}