leetcode 904 水果成篮

一、题目描述

二、解题思路

题意解析：

fruit[i]是指i位置树上果实的种类，例如fruits[0]为1，代表0位置树上的是1号果实。只有两个篮子，每个篮子只能装同种果实，比如只能装1号、2号等等。一棵树只能摘一个水果，题目要求最多可以摘的水果数量。

整体思路：

题目要求最多可以摘的水果数量。也就是求只有两种类型的最长连续子串的长度，我们可以借助滑动窗口的思想来解决这个问题。

具体思路：

哈希表nums用于记录每一种类型的水果采摘的个数，用于模拟篮子。

<1>进窗口：进入哈希表nums

//进窗口

nums[fruits[right]]++;

<2>出窗口：退出哈希表，left右移

//出窗口

while(nums.size()>2){

nums[fruits[left]]--;

if (nums[fruits[left]]==0) nums.erase(fruits[left]);

left++;

}

<3>更新：更新length

//更新

length=max(length,right-left+1);

三、代码实现

时间复杂度：T(n)=O(n)

空间复杂度：S(n)=O(1)

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        //滑动窗口
        int left,right,n=fruits.size();
        map<int,int> nums;
        int length=0;
        for(left=0,right=0;right!=n;right++){
            //进窗口
            nums[fruits[right]]++;
            //出窗口
            while(nums.size()>2){
                nums[fruits[left]]--;
                if (nums[fruits[left]]==0)  nums.erase(fruits[left]);
                left++;
            }
            //更新
            length=max(length,right-left+1);
        }
        return length;
    }
};