LeetCode 刷题 904

思考这一题时 苦恼不知道怎么比较好的表达已经遍历过的元素的种类(不同的数), 也就是题目说的水果种类。 于是看了下参考答案, 用的是unordered_map 还没有学习到,于是学了稍微学了下。

unordered_map 是无序关联容器, 是对hash_map的实现,对unordered进行查找、输入、删除是常数时间复杂度。

  1. 将不同的“水果”看成不同的键, 遍历到一种“水果“就将该键所对应的值+1, 然后++fast(这前提是map的size不能大于2)

  1. 当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() 返回该键值对的迭代器。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值