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

 977. 有序数的平方

参考文章：代码随想录 (programmercarl.com)

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> ans(nums.size(), 0);
        int k = nums.size() - 1;
        int j = nums.size() - 1;
        int i = 0; 
        while (i <= j || k >= 0) {
            if (nums[i] * nums[i] < nums[j] * nums[j]) {
                ans[k--] = nums[j] * nums[j];
                j--;
            }
            if (nums[i] * nums[i] >= nums[j] * nums[j]){
                ans[k--] = nums[i] * nums[i];
                i++;
            }
        }
        return ans;
    }
};

一开始是没有想到这个方法的。其实最直接的想法是直接把有数组平方然后再排序。

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

不过这样有些偷鸡了😛。

209. 长度最小的子数组

参考文章：代码随想录 (programmercarl.com)

也是一开始没什么思路。看了卡哥的解法。

class Solution {
public:
    int minSubArrayLen(int target, vector<int>& nums) {
        int ans = INT_MAX;
        int j = 0;
        int len = 0;
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) {
            sum += nums[i]; 
            while(sum >= target) {
                len = i - j + 1;
                ans = ans < len ? ans : len;
                sum -= nums[j];
                j++;
            }
        }
        return ans < INT_MAX ? ans : 0;
    }
};

59.螺旋矩阵

暴力模拟，但要注意边界。

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> ans(n, vector(n, 0));
        int l = 0, r = n - 1;
        int u = 0, d = n - 1;
        int count = 1;
        while (count <= n * n) {
            for (int i = l; i <= r; i++, count++) {
                ans[u][i] = count;
            }
            ++u;
            for (int i = u; i <= d; i++, count++) {
                ans[i][r] = count;
            }
            --r;
            for (int i = r; i >= l; i--, count++) {
                ans[d][i] = count;
            }
            --d;
            for (int i = d; i >= u; i--, count++) {
                ans[i][l] = count;
            }
            ++l;
        }
        return ans;
    }
};

总结

今天也是收获满满，加深了算法的理解。