代码随想录 Day 2-977.有序数组的平方 ,209.长度最小的子数组 ,59.螺旋矩阵II

141 阅读2分钟

Day 2-977.有序数组的平方 ,209.长度最小的子数组 ,59.螺旋矩阵II

977.有序数组的平方

暴力排序(初始思路)

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> ret;
        for(int i = 0; i < nums.size(); i++){
            ret.push_back(nums[i] * nums[i]);
        }
        std::sort(ret.begin(), ret.end());
        return ret; 
    }
};

快排的时间复杂度是O(nlogn)

双指针法

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

学习一下这个for循环的写法

209.长度最小的子数组

初始思路(超出时间限制)

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int ret = 100005;
        for(int i = 0; i < nums.size(); i++){
            int sum = 0;
            for(int j = i; j < nums.size(); j++){
                sum += nums[j];
                if(sum >= target){
                    if(j - i + 1 < ret){
                        ret = j - i + 1;
                    }
                    break;
                }
            }
        }
        if(ret == 100005) return 0;
        return ret;
    }
};

滑动窗口

也可以理解为双指针的一种

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. 螺旋矩阵

初始化vector

image-20230203191956201

vector<int> vec1(7);
vector<int> vec2(7, 3);

初始解法

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> vec;
        for(int i = 0; i < n; i++){
            vector<int> a(n);
            vec.push_back(a);
        }
        int i = 0;
        int j = 0;
        int up = 1;
        int left = 0;
        int down = n - 1;
        int right = n - 1;
        int direction = 0;
        for(int k = 1; k < n * n + 1; k++){
            vec[i][j] = k;
            switch(direction){
                case 0:
                    if(j < right){
                        j ++;
                    }else{
                        i ++;
                        right --;
                        direction ++;
                    }
                    break;
                case 1:
                    if(i < down){
                        i ++;
                    }else{
                        j --;
                        down --;
                        direction ++;
                    }
                    break;
                case 2:
                    if(j > left){
                        j --;
                    }else{
                        i --;
                        left ++;
                        direction ++;
                    }
                    break;
                case 3:
                    if(i > up){
                        i --;
                    }else{
                        j ++;
                        up ++;
                        direction = 0;
                    }
                    break;
                default:
                    break;
            }
        }
        return vec;
    }
};

示例

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 = 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 += 1;
        }
​
        // 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
        if (n % 2) {
            res[mid][mid] = count;
        }
        return res;
    }
};

抓住不变量,判断好闭区间和开区间

vector<vector<int>> res(n, vector<int>(n, 0)); 如何创建二维数组