数组篇——双指针法(leetcode209)滑动窗口法(leetcode904)

文章讲述了如何使用双指针法解决LeetCode中的两个问题:找到和大于等于给定值的最小连续子数组长度,以及在有限篮子条件下最大化果树采摘水果数。介绍了两种方法,一种是个人无脑写法,另一种是利用哈希表优化的空间复杂度较低的方法。
摘要由CSDN通过智能技术生成

 1.[leetcode209]

给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的 连续 子数组,并返回其长度。如果不存在符合条件的子数组,返回 0。

示例:

  • 输入:s = 7, nums = [2,3,1,2,4,3]
  • 输出:2
  • 解释:子数组 [4,3] 是该条件下的长度最小的子数组。

时间复杂度:O(n)

空间复杂度:O(n)

原理:新数组最大元素一定由原数组最左端或最右端元素平方得到,于是在两端比较后得到新数组最后一个元素,然后移动左右指针继续比较

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
      //双指针法:新数组的最大值一定在两端
      int i=0,j=nums.size()-1,k=j;
      vector<int> result(j+1,0);
      while(i<=j)
      {
          if(nums[i]*nums[i]>nums[j]*nums[j]) {
              result[k--]=nums[i]*nums[i];
              i++;
          }
          else {
             result[k--]=nums[j]*nums[j]; 
              j--;
          } 
      }
      return result;
    }
};

2. [leetcode 904]

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

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

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

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

示例 1:

输入:fruits = [1,2,1]
输出:3
解释:可以采摘全部 3 棵树。

示例 2:

输入:fruits = [0,1,2,2]
输出:3
解释:可以采摘 [1,2,2] 这三棵树。
如果从第一棵树开始采摘,则只能采摘 [0,1] 这两棵树。

示例 3:

输入:fruits = [1,2,3,2,2]
输出:4
解释:可以采摘 [2,3,2,2] 这四棵树。
如果从第一棵树开始采摘,则只能采摘 [1,2] 这两棵树。

解法一:个人无脑写法,原理很简单,当种类超过2时向左压缩区间 

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int type1=-1,type2=-1; //两种篮子对应的水果类型
        int type1_num=0,type2_num=0; //两种水果类型当前装的水果数量
        int low=0,high; //设置滑动窗口的左右边界
        int ans=0;
        //移动窗口右边界(摘果子)
        for(high=0;high<fruits.size();high++)
        {
            //一号篮装水果
            if(fruits[high]==type1) {
                type1_num++;
            }
            //二号篮装水果
            else if(fruits[high]==type2) {
                type2_num++;
            }
            else{
            //下讨论水果种类与篮中水果不同情况
            //如果两个果篮都是空的,一号篮装上水果
            if(type1==-1&&type2==-1){
            type1=fruits[high];
            type1_num++;
            }
            //如果一号果篮不空并且二号果篮空,二号篮装上水果
            else if(type2==-1) {
                type2=fruits[high];
                type2_num++;
            } 
             //如果二号果篮不空并且一号果篮空,一号篮装上水果
            else if(type1==-1) {
                type1=fruits[high];
                type1_num++;
            } 
            //如果新出现第三类水果,则记录当前子数组长度并移动左窗口压缩区间
            else{
                if(high-low>ans)
                ans=high-low;
                //当两个篮子都不为空时继续压缩
                while(type1_num!=0&&type2_num!=0)
                {
                      if(fruits[low]==type1) type1_num--;
                      else type2_num--;
                      low++;   
                }
                //把新水果装入空出的篮子中
                if(type1_num==0){
                    type1=fruits[high];
                    type1_num++;
                }
                else {
                    type2=fruits[high];
                    type2_num++;
                }

            }

            }

        }
    //别忘了最后答案更新
         ans=high-low>ans?high-low:ans;
        return ans;
    }
};

 下面是官方使用哈希表的答案

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int n = fruits.size();
        unordered_map<int, int> cnt; //哈希表 

        int left = 0, ans = 0;
        for (int right = 0; right < n; ++right) {
            ++cnt[fruits[right]]; //添加水果种类
            while (cnt.size() > 2)//水果种类大于2时 {
                auto it = cnt.find(fruits[left]);
                --it->second;
                if (it->second == 0) {
                    cnt.erase(it);
                }
                ++left;
            }
            ans = max(ans, right - left + 1);
        }
        return ans;
    }
};

时间复杂度:O(n)

空间复杂度:O(1)哈希表中最多会有三个键值对,可以看成使用了常数级别的空间。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值