代码随想录算法训练营第二天| 977.有序数组的平方，209.长度最小的子数组，59.螺旋矩阵II

最新推荐文章于 2024-07-18 20:37:52 发布

来陪可莉炸鱼吧

最新推荐文章于 2024-07-18 20:37:52 发布

阅读量23

点赞数

分类专栏： LeetCode 文章标签：算法矩阵排序算法

本文链接：https://blog.csdn.net/Sheldon_2020/article/details/132642759

版权

LeetCode 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

977.有序数组的平方

给你一个按非递减顺序排序的整数数组 nums，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

示例 1：

输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]，排序后，数组变为 [0,1,9,16,100]

示例 2：

输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

暴力法

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;
    }
};

时间复杂度O(n+nlogn)

双指针法

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

时间复杂度为O(n)

209.长度最小的子数组

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

示例：

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

提示：

1 <= target <= 10^9
1 <= nums.length <= 10^5
1 <= nums[i] <= 10^5

暴力法

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;
    }
};

滑动窗口

窗口内是什么？
如何移动窗口的起始位置？
如何移动窗口的结束位置？

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

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

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

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

时间复杂度：O(n)

不要以为for里放一个while就以为是O(n^2)啊，主要是看每一个元素被操作的次数，每个元素在滑动窗后进来操作一次，出去操作一次，每个元素都是被操作两次，所以时间复杂度是 2 × n 也就是O(n)。

59.螺旋矩阵II

给定一个正整数 n，生成一个包含 1 到 n^2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]

key:坚持循环不变量原则(左闭右开）

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

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

            // 模拟填充右列从上到下(左闭右开)
            for(i=startx;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++;
        }

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