算法训练 | 数组Part2 | 977.有序数组的平方、209.长度最小的子数组、59.螺旋矩阵II

目录

977.有序数组的平方

暴力解法

双指针法

209.长度最小的子数组

暴力解法

滑动窗口

59.螺旋矩阵II

转圈模拟

---

嵌入式学习分享个人主页：Orion嵌入式随想录 - 小红书 (xiaohongshu.com)

977.有序数组的平方

• 题目链接：. - 力扣（LeetCode）

• 文章讲解：代码随想录

• 视频讲解： 双指针法经典题目 | LeetCode：977.有序数组的平方_哔哩哔哩_bilibili

暴力解法

• 解题思路

  • 先平方后排序

• 解题步骤

  • 循环内平方

  • 循环后快速排序

• 代码注意

  • sort(A.begin(), A.end());快速排序方法

  • return nums;最后返回数组本身

• 代码一：暴力排序

//时间复杂度 O(n + nlogn) O(nlogn)
class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        for (int i = 0; i < A.size(); i++) {
            A[i] *= A[i];
        }
        sort(A.begin(), A.end()); // 快速排序
        return A;
    }
};

双指针法

• 解题思路

  • 数组是有序的， 只不过负数平方之后可能成为最大数了。

  • 则数组平方的最大值就在数组的两端，不可能是中间。

  • 此时可以考虑双指针法，i指向起始位置，j指向终止位置。

• 解题步骤

  • 定义新数组result

  • 定义索引下标k，最大值开始更新

  • 循环两个参数ij，循环条件i<=j

  • 判断两端值大小，result索引值更新，索引k--

  • 根据判断结果执行i++或j--

• 代码注意

  • vector<int> result(A.size(), 0);动态数组创建

  • k = A.size() - 1;索引角标

  • (int i = 0, j = A.size() - 1; i <= j;) j前无int，条件后有“；”。此处不更新i与j

• 代码二：

//时间复杂度为O(n)
class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        int k = A.size() - 1;
        vector<int> result(A.size(), 0);
        for (int i = 0, j = A.size() - 1; i <= j;) { // 注意这里要i <= j，因为最后要处理两个元素
            if (A[i] * A[i] < A[j] * A[j])  {
                result[k--] = A[j] * A[j];
                j--;
            }
            else {
                result[k--] = A[i] * A[i];
                i++;
            }
        }
        return result;
    }
};

209.长度最小的子数组

• 题目链接：209. 长度最小的子数组 - 力扣（LeetCode）

• 文章讲解：programmercarl.com

• 视频讲解：拿下滑动窗口！ | LeetCode 209 长度最小的子数组_哔哩哔哩_bilibili

• 904.水果成篮(opens new window)

• 76.最小覆盖子串

暴力解法

• 解题思路

  • 两个for循环，一个for循环滑动窗口的起始位置，一个for循环为滑动窗口的终止位置，不断的寻找符合条件的子序列，时间复杂度是O(n^2)。

• 解题步骤

  • 初始化最大可能值

  • 外循环起点从0开始

  • 外循环：每次需清除sum

  • 内循环终点从起点开始

  • 内循环：sum更新，超过s则停止更新sum，找到长度并返回跳出

  • 可能值没有被更新则返回0

• 代码注意

  • result = result < subLength ? result : subLength; 是subLength不是nums.size()，

  • sum += nums[j];更新sum

• 代码一：暴力循环

// 时间复杂度O(n^2) 空间复杂度：O(1)
class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        int result = INT32_MAX; // 最终的结果
        int sum = 0; // 子序列的数值之和
        int subLength = 0; // 子序列的长度
        for (int i = 0; i < nums.size(); i++) { // 设置子序列起点为i
            sum = 0;
            for (int j = i; j < nums.size(); j++) { // 设置子序列终止位置为j
                sum += nums[j];
                if (sum >= s) { // 一旦发现子序列和超过了s，更新result
                    subLength = j - i + 1; // 取子序列的长度
                    result = result < subLength ? result : subLength;
                    break; // 因为我们是找符合条件最短的子序列，所以一旦符合条件就break
                }
            }
        }
        // 如果result没有被赋值的话，就返回0，说明没有符合条件的子序列
        return result == INT32_MAX ? 0 : result;
    }
};

滑动窗口

• 滑动窗口的核心思想是在数据集合中维护一个窗口，随着遍历的进行，窗口的大小和位置可以动态调整，以此来解决特定问题。

• 滑动窗口的实现通常涉及两个指针（或索引），一个指向窗口的开始，另一个指向窗口的结束。随着遍历的进行，可以根据需要移动这些指针来扩展或收缩窗口。

• 就是不断的调节子序列的起始位置和终止位置，从而得出结果。

• 解题思路

  • 如果用一个for循环，那么应该表示 滑动窗口的起始位置，还是终止位置。

  • 如果只用一个for循环来表示 滑动窗口的起始位置，那么如何遍历剩下的终止位置？

  • 所以只用一个for循环，那么这个循环的索引，一定是表示滑动窗口的终止位置。

  • 窗口就是 满足其和 ≥ s 的长度最小的 连续 子数组。

  • 窗口的起始位置如何移动：如果当前窗口的值大于等于s了，窗口就要向前移动了（也就是该缩小了）。

  • 窗口的结束位置如何移动：窗口的结束位置就是遍历数组的指针，也就是for循环里的索引。

• 解题步骤

  • 初始化最大可能值

  • 外循环终点从0开始

  • 外循环：更新sum，判断是否符合

  • 符合则找到长度，更新sum并移动起始位置再次判断

  • 可能值没有被更新则返回0

• 代码注意

  • return result == INT32_MAX ? 0 : result;只返回一次，注意返回位置在循环外

• 代码二：滑动窗口

// 时间复杂度O(n)
class Solution {
public:
    int minSubArrayLen(int s, vector<int>& nums) {
        int result = INT32_MAX;
        int sum = 0; // 滑动窗口数值之和
        int i = 0; // 滑动窗口起始位置
        int subLength = 0; // 滑动窗口的长度
        for (int j = 0; j < nums.size(); j++) {
            sum += nums[j];
            // 注意这里使用while，每次更新 i（起始位置），并不断比较子序列是否符合条件
            while (sum >= s) {
                subLength = (j - i + 1); // 取子序列的长度
                result = result < subLength ? result : subLength;
                sum -= nums[i++]; // 这里体现出滑动窗口的精髓之处，不断变更i（子序列的起始位置）
            }
        }
        // 如果result没有被赋值的话，就返回0，说明没有符合条件的子序列
        return result == INT32_MAX ? 0 : result;
    }
};

59.螺旋矩阵II

• 题目链接：. - 力扣（LeetCode）

• 文章讲解：代码随想录

• 视频讲解：一入循环深似海 | LeetCode：59.螺旋矩阵II_哔哩哔哩_bilibili

• 54.螺旋矩阵(opens new window)

• 剑指Offer 29.顺时针打印矩阵

转圈模拟

• 二分法的衍生，保持循环边界不变原则。模拟顺时针画矩阵的过程。

• 解题思路

  • 填充上行从左到右

  • 填充右列从上到下

  • 填充下行从右到左

  • 填充左列从下到上

• 解题步骤

  • 定义起始边界

  • 设置转圈数为n/2，奇数单独填充处理。

  • 设置填充值和缩进值

  • 有圈数剩余则循环

  • 循环内顺时针填充，保持边界一致性

  • 转完一圈，边界索引++，缩进值++

  • 补充奇数情况count（n / 2，n / 2）

• 代码注意

  • vector<vector<int>> res(n, vector<int>(n, 0));二维数组定义

  • int i,j;不赋值0 ，不是从0开始，每次循环要赋值x,y

• 代码一：左闭右开

// 时间复杂度 O(n^2): 模拟遍历二维矩阵的时间
// 空间复杂度 O(1)
class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
        int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
        int loop = n / 2; // 每个圈循环几次，例如n为奇数3，那么loop = 1 只是循环一圈，矩阵中间的值需要单独处理
        int mid = n / 2; // 矩阵中间的位置，例如：n为3， 中间的位置就是(1，1)，n为5，中间位置为(2, 2)
        int count = 1; // 用来给矩阵中每一个空格赋值
        int offset = 1; // 需要控制每一条边遍历的长度，每次循环右边界收缩一位
        int i,j;
        while (loop --) {
            i = startx;
            j = starty;

            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上行从左到右(左闭右开)
            for (j; j < n - offset; j++) {
                res[i][j] = count++;
            }
            // 模拟填充右列从上到下(左闭右开)
            for (i; i < n - offset; i++) {
                res[i][j] = count++;
            }
            // 模拟填充下行从右到左(左闭右开)
            for (; j > starty; j--) {
                res[i][j] = count++;
            }
            // 模拟填充左列从下到上(左闭右开)
            for (; i > startx; i--) {
                res[i][j] = count++;
            }

            // 第二圈开始的时候，起始位置要各自加1， 例如：第一圈起始位置是(0, 0)，第二圈起始位置是(1, 1)
            startx++;
            starty++;

            // offset 控制每一圈里每一条边遍历的长度
            offset += 1;
        }

        // 如果n为奇数的话，需要单独给矩阵最中间的位置赋值
        if (n % 2) {
            res[mid][mid] = count;
        }
        return res;
    }
};

（说明：基于代码随想录课程学习，部分内容引用自代码随想录文章）