【代码随想录】算法训练营 第二天 第一章 数组 Part 2

977. 有序数组的平方

题目

暴力解法

思路

原地更新所有数组元素为其平方数后，再使用sort函数排序，对vector使用sort函数时，两个参数分别是vector的起始元素和终止元素。

代码

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

双指针法

思路

从题目中我们可以知道，数组元素有正有负，从小到大排序，但是平方以后，元素大小的分布就变成了两边大中间小；

基于此，我们使用双指针法来代替sort函数，双指针在归并排序里就用到了，但在这里，两个指针不是指着两个数组，而是指着输入数组的两端，每次比较两端元素的大小，将较大元素放在新数组里（新数组用vector定义一个和原数组内容一样的，新数组的指针每次向左移动一位），然后指针向内移动一位，直到两个指针相遇。

代码

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

209. 长度最小的子数组