思考这一题时 苦恼不知道怎么比较好的表达已经遍历过的元素的种类(不同的数), 也就是题目说的水果种类。 于是看了下参考答案, 用的是unordered_map 还没有学习到,于是学了稍微学了下。
unordered_map 是无序关联容器, 是对hash_map的实现,对unordered进行查找、输入、删除是常数时间复杂度。
将不同的“水果”看成不同的键, 遍历到一种“水果“就将该键所对应的值+1, 然后++fast(这前提是map的size不能大于2)
当size > 2时, 就说明应该抛弃前面slow所代表的水果, 当然 随着一个个抛弃,两个种类其中一种会先被丢光, 此时size就会等于2, 就可以继续遍历数组
class Solution {
public:
int totalFruit(vector<int>& fruits) {
unordered_map<int, int> fruitMap;
int slow = 0;
int fast = 0;
int ans = 0;
while (fast != fruits.size())
{
++fruitMap[fruits[fast]];
while (fruitMap.size() > 2)
{
auto it = fruitMap.find(fruits[slow]);
--it->second;
if (it->second == 0)
fruitMap.erase(it);
++slow;
}
ans = fast - slow + 1 > ans ? fast - slow + 1 : ans;
++fast;
}
return ans;
}
};
fruitMap.find() 返回该键值对的迭代器。