代码随想录算法训练营第二天 | 977. 有序数组的平方、27.移除元素

LeetCode 977.有序数组的平方

链接：https://leetcode.cn/problems/squares-of-a-sorted-array/

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]){
                result[k--] = nums[i]*nums[i];
                i++;
            }
            else{
                result[k--] = nums[j]*nums[j];
                j--;
            }
        }
        return result;
    }
};

心得：

1. 没有想到利用例子中数组元素的特性，它是由负到正的非递减数组，因为新数组是按元素平方来进行排序，所以没有必要用暴力方式排序，可以从两端最大的开始挑选（比较绝对值），并且数组赋值也从后往前。否则如果按顺序排序，那么数组从负数到0绝对值是递减的，而从0到正数时绝对值又是递增的，这样很麻烦。

 

59.螺旋矩阵Ⅱ

链接：https://leetcode.cn/problems/spiral-matrix-ii/

 

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> result(n, vector<int>(n, 0));
        int startx = 0, starty = 0, offset = 1, count = 1;
        int loop = n/2;
        int mid = n/2;
        int i, j;
        while(loop--){
            for(i = startx; i < n-offset; i++) result[starty][i] = count++;
            for(j = starty; j < n-offset; j++) result[j][i] = count++;
            for(; i > startx; i--) result[j][i] = count++;
            for(; j > starty; j--) result[j][i] = count++;
            startx++;
            starty++;
            offset++;
        }
        if(n%2 == 1) result[mid][mid] = n*n;
        return result;
    }
};

心得：

1. 我在思考的时候只考虑了行和列，也就是分成两种循环，所以一直处理不好逻辑；考虑成加方向的行和列，也就是四种，会更好操作一点。还有就是当n是奇数的时候，中间单独的一个元素应该特别处理，这一点我没有考虑到。