Leetcode刷题笔记——904. 水果成篮【哈希表、滑动窗口】

题目

思路

理解完题目即可发现，这题的本质是要求数组中的最长连续子数组长度，不同的是这个连续子数组可以拥有两个不同的元素。由于存在不同的元素，首先可以想到用哈希表记录子数组不同元素的出现次数，该哈希表的长度应该不超过2，一旦超过2说明出现了第3种元素，需要将之前出现的2种元素中的任意一种剔除出去，这种右侧向右扩展，出现意外需要左侧运动的思路很容易想到滑动窗口。

滑动窗口模板代码：

int left=0;
int len=0,res=0;
for(int i=0;i<nums.size();i++){
	len++;
	//对右侧新元素要做的操作
	...
	//
	while(...){// 新加入右侧元素后，判断整个滑动窗口的内容是否符合要求,不符合要求则进入while
	...	//当前滑动窗口内容不符合要求，对滑动窗口左侧内容进行修改，直到符合要求
	len--;
	left++;//对滑动窗口左侧的内容进行修改的同时一般都伴随着len--、left++等操作
	}
	if(len>res){
		res = len;
	}
	return res;

代码

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        unordered_map<int,int> hashmap;
        int res = 0,len = 0;
        int left = 0;
        for(int i=0;i<fruits.size();i++){
            hashmap[fruits[i]]++;
            len++;
            while(hashmap.size()>2){
                hashmap[fruits[left]]--;
                // 如果哈希表中左侧元素出现的次数为0，则需要擦除这个元素
                if(hashmap[fruits[left]]==0) hashmap.erase(fruits[left]);
                left++;
                len--;
            }
            if(len>res){
                res=len;
            }
        }
        return res;
    }
};