[C++学习] 数组滑动窗口代码随想录 LeetCode题代码笔记

最新推荐文章于 2024-09-21 17:52:59 发布

今天也还活着

最新推荐文章于 2024-09-21 17:52:59 发布

阅读量518

点赞数 14

分类专栏： C++学习文章标签： c++ 学习 leetcode

本文链接：https://blog.csdn.net/qq_49696822/article/details/135772589

版权

C++学习专栏收录该内容

9 篇文章 1 订阅

订阅专栏

本文介绍了如何使用滑动窗口法解决LeetCode中的两道题目，分别是寻找总和大于等于目标值的长度最小连续子数组（209题）和寻找只有两个不同元素的最长连续子数组（904题）。通过实例展示了如何运用双指针技巧调整窗口起始和终止位置。

摘要由CSDN通过智能技术生成

滑动窗口法：不断调节子序列的起始位置和终止位置，其实是双指针，但看起来像是一个窗口在不断滑动。

一、LeetCode题 209.长度最小的子数组 209. 长度最小的子数组 - 力扣（LeetCode）

题目描述：总和大于等于，长度最小，连续子数组

窗口：满足其和大于等于目标值长度最小的连续子数组。

窗口起始位置：当当前窗口值大于目标值之后 , 需要向前移动

窗口终止位置：遍历数组的指针 , 即for循环的索引

代码：

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int result = INT_MAX;
        int i = 0; // 滑动窗口起始位置
        int sum = 0; //当前窗口内元素的总和
        int subLength = 0; //连续子数组的长度
        for (int j = 0; j < nums.size(); j++) {
            // j 为滑动窗口的终止位置
            sum += nums[j];
            while (sum >= target) {
                subLength = j - i + 1; //保存符合条件的子数组的长度
                result = min(result, subLength); //判断当前子数组是否长度最小
                sum -= nums[i]; //关键：更新滑动窗口起始位置
                i++;
            }
        }
        return result == INT_MAX ? 0 :result;
    }
};

1、INT_MAX 和 INT_MIN 是 C++ 中的两个宏，代表了整型变量能够存储的最大正整数和最小负整数，分别为 2147483647 和 -2147483648，在头文件 <limits.h> 中定义。

2、三目运算符：1 ？ 2 ： 3 若1为真，执行返回2，若1为假，执行返回3。

二、LeetCode题 904. 水果成篮 904. 水果成篮 - 力扣（LeetCode）

题目描述：寻找只有两个不同元素的最长连续子数组

代码：关键是想清窗口起始位置和终止位置的更新规则。

class Solution {
public:
    int totalFruit(vector<int>& fruits) {
        int i = 0;         //滑动窗口起始位置
        int result = 0;    //能够采集水果的最大数目
        int subLength = 0; //保存每个窗口内采集水果的数目
        int right = 0, left = 0, k = 0;
        if(fruits.size() < 3) return fruits.size();
        for (int j = 0; j < fruits.size(); j++) {
            
            for(k = i; k < fruits.size(); k++){
                // 找到第二个水果种类，保存索引和大小
                if(fruits[k] != fruits[i]){
                    right = fruits[k];
                    left = k;
                    break;
                }
                // 整个数组只有一个数的情况
                else if(k == fruits.size() - 1) return k - i + 1;
            }
            while(fruits[k] == fruits[i] || fruits[k] == right){
                // 当种类符合时，不断向右滑动窗口
                subLength = k - i + 1;
                result = max(result, subLength);
                if(k == fruits.size() - 1) return result;
                k++;
            }
            // 种类不符合时，更新起始位置至保存的第二个水果种类起始位置
            i = left;
        }
        return result;
    }
};