水果成篮（c++）

题目描述

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果 种类 。

你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：

• 你只有 两个 篮子，并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
• 你可以选择任意一棵树开始采摘，你必须从 每棵 树（包括开始采摘的树）上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次，你将会向右移动到下一棵树，并继续采摘。
• 一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。

给你一个整数数组 fruits ，返回你可以收集的水果的 最大 数目。

解决思路

考虑滑动窗，用两个int记录，右指针持续移动，不满足条件时（即不等于两种类别时）记录上一段长度，并处理左指针进入新的窗口，从当前右端点位置倒序寻找符合条件的端点，然后重新记录类别索引。重复循环到最后。

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int i = 0, j = 0;
        int c1 = 0, c2 = 0; //类别起始索引，c1前类，c2后类
        int num = 0;
        for(; i < fruits.size(); ++i){
            if(fruits[i] != fruits[c1] and fruits[i] != fruits[c2]){
                if(c1 == c2){ 
                    c2 = i; //初始类别设置相同，寻找到新类别时索引赋给c2
                }
                else{
                    c2 = i; //新类别赋给c2
                    c1 = i - 1; //新类别前一个元素一定为另一个类别
                    num = i - j > num ? i - j : num; // 比较上一段窗口与历史最长，保存最长
                    j = i; // 从新类别前一个位置开始寻找左端点
                    while(fruits[j-1] == fruits[c2] or fruits[j-1] == fruits[c1]){
                        --j;
                    }
                    c1 = j;
                }
            }
        }
        return max( i - j , num ); 
    }
};