数组NO8:904. 水果成篮

1.题目

你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示,其中 fruits[i] 是第 i 棵树上的水果 种类 。
你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必须按照要求采摘水果:
你只有 两个 篮子,并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
你可以选择任意一棵树开始采摘,你必须从 每棵 树(包括开始采摘的树)上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次,你将会向右移动到下一棵树,并继续采摘。
一旦你走到某棵树前,但水果不符合篮子的水果类型,那么就必须停止采摘。
给你一个整数数组 fruits ,返回你可以收集的水果的 最大 数目。
示例 1:
输入:fruits = [1,2,1]
输出:3
解释:可以采摘全部 3 棵树。
示例 2:
输入:fruits = [0,1,2,2]
输出:3
解释:可以采摘 [1,2,2] 这三棵树。
如果从第一棵树开始采摘,则只能采摘 [0,1] 这两棵树。

2.解法一

2.1. 思路分析

首先读懂问题,理解问题的本质。此问题的本质就是求最长连续子序列的长度。
要解决此问题,首先想到的就是滑动窗口法,而要使用滑动窗口法,就要解决三个问题:
-窗口的内容是什么?
-窗口的右端何时更新?
-窗口的左端何时更新?更新的时候窗口的内容要如何更新?

2.2. 算法描述

-首先,创建窗口、待求量、计数器;
-然后,当窗口内的元素个数小于两个时,计数器自增;
-然后,当元素大于两个时,计数器自减,窗口内容更新,直至count再次小于2
-最后等窗口右端到达数组边界,终止循环,返回length。

2.3. 算法实现

int totalFruit(int* fruits, int fruitsSize){
    int left = 0;
    int right = 0;
    int count = 0;
    int length = 0;
    int hash[100001] = {0};

    while (right < fruitsSize) {
        if (hash[fruits[right]] == 0) {
            count++;
        }
        hash[fruits[right]]++;

        if (count <= 2) {
            length = fmax(length, right - left + 1);
        }

        while (count > 2 && left < fruitsSize) {
            hash[fruits[left]]--;

            if (hash[fruits[left]] == 0) {
                count--;
            }
            left++;
        }
        right++;
    }

    return length;
}

2.4.算法分析

此算法的时间复杂度为O(n)?

3.总结

3.1.滑动窗口算法的模板

/ 模板
while () {
    // 将新进来的右边的数据,计算进来
    // 更新数据

    // 判断窗口数据是否不满足要求了
    while (窗口数据不满要求 && left < arrSize) {
        // 移除left数据,更新窗口数据
        left++;
    }
    right++;
}

3.2.hash表数据结构

  • 在本题中需要有一种数据结构来记录每个元素出现的次数,hash表最为合适,以值为键,以出现的次数为值,后面搜索的时候时间复杂度为O(1)。
  • 由于本题中的数据规模较小,因此使用数组来模拟hash表。
posted @ 2021-12-20 20:04  曼路的个人博客  阅读(45)  评论(0编辑  收藏  举报