水果成篮. - 备战技术面试?力扣提供海量技术面试资源,帮助你高效提升编程技能,轻松拿下世界 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;
}
}