leetcode904. 水果成篮(滑动窗口)

一:题目

二:上码

class Solution {
public:

    /**
        思路:滑动窗口
            1.滑动窗口为我们 两个品种水果的个数
            2.滑动窗口的起始位置为数组的起始位置,当遇见一个新的水果种类的时候,这时候就需要移动我们的起始位置
                这里移动起始位置是要注意的,我们需要移动起始到我们的窗口中 完全没有 这个品种为止
            3.滑动窗口的终止位置为数组下标
    */
    int totalFruit(vector<int>& fruits) {
        
        int length = 0;//窗口的大小
        int i = 0;//窗口的起始位置
        int ans = 0;
        map<int,int>m;//key值变表示苹果的种类 ，value 表示该品种的数量  
        map<int,int>::iterator mt;

        for (int j = 0; j < fruits.size(); j++) {            
            m[fruits[j]]++;//每个苹果的种类加一
            
            while(m.size() > 2) {//需要开始移动我们的起始位置

                length = j-i;
                ans = ans < length ? length:ans;// 这里的j不用加一，因为我们是统计到第三种水果时进的while
        
        
                m[fruits[i]]--;//第一种水果的个数递减

                if(m[fruits[i]] == 0) {//当其个数为0的时候;我们就要移除此元素
                    m.erase(fruits[i]);
                }        
                
                i++;//他会把篮子中的第一种水果 慢慢 耗光,剩下的是我们的第紧贴着的新品种水果
                    //1 1 2 1 2 3 会把1慢慢耗光
            }   
        
        }

        //处理水果种类 小于2种  当然即便水果种类大于3 我们最终也是剩下两种,我们也是可以对其进行统计的,只不过sum
        if(m.size() <= 2) {                                                               //ans值相等。
            int sum = 0;
            for (mt = m.begin(); mt != m.end(); mt++) {
                sum+=mt->second;
            }
            ans = ans < sum ?sum :ans;
        }

        return ans;
    }
};