904. 水果成篮

最新推荐文章于 2024-11-11 20:59:17 发布

霜白sakura

最新推荐文章于 2024-11-11 20:59:17 发布

阅读量89

点赞数

分类专栏： leetcode 文章标签：双指针算法 LeetCode 904.水果成篮 C++ 动态规划

本文链接：https://blog.csdn.net/weixin_44458954/article/details/127358252

版权

leetcode 专栏收录该内容

58 篇文章 0 订阅

订阅专栏

题目：904. 水果成篮
思路：双指针

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int n = fruits.size();
        if(n==1) return 1;
        int le=0,ri=1;
        int res = 0;
        while(le<n && ri<n){
            int fi = fruits[le];
            int se = -1;
            while(ri<n){
                if(fruits[ri]==fi){
                    ri++;
                }else if(se == -1){
                    se = fruits[ri];
                    ri++;
                }else if(fruits[ri]==se){
                    ri++;
                }else{
                    break;
                }
            }
            res = max(res,ri-le);
            if(ri>=n) break;
            //cout<<le<<" 5  "<<ri<<endl;
            int newle = 0;
            for(int i=ri-1;i>le;i--){
                if(fruits[i]==fruits[ri-1]){
                    newle = i;
                    continue;
                }else{
                    newle = i+1;
                    break;
                }
            }
            le=newle;
            ri=le+1;
            //cout<<le<<" 4 "<<ri<<endl;
        }
        return res;
    }
};