力扣3：有序数组的平方

977. 有序数组的平方 - 力扣（LeetCode）

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

示例 1：

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

解法1：暴力解法

先挨个平方，再用sort函数排序

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

注意：定义函数的时候，函数类型是vector<int>类型，要返回一个数组

解法2：双指针

分析一下题目不难看出，数组本身是非递减的，就算平方之后，最大值也是在两端，要么最左边，要么最右边，越往中间越小，那么就可以定义两个指针，分别在数组两端，新建一个数组，定义新的指针k，放在数组最右端，原数组的两个指针比较后，大的元素往k指针处填，每填一个，k向左移动一格，直到比较完所有元素。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums)
 {
    vector<int> result(nums.size(),0);
    int k=nums.size()-1;
    for(int i=0,j=nums.size()-1;i<=j;)  //注意这个i要小于等于j。
    {
        if(nums[i]*nums[i]>nums[j]*nums[j])
        {
            result[k]=nums[i]*nums[i];
            k--;
            i++;
        }
        else
        {   
            result[k]=nums[j]*nums[j];
            k--;
            j--;

        }
        
    }
    return result;
 } 
};

注意：在for循环的比较过程中，满足下一次的循环条件是i小于等于j，如果不加等于，当数组两两比较完，两个指针指向剩下的最后一个元素时，i是等于j的，这个时候不满足i<j这个条件，就直接return数组了，会漏掉最后一个最小的元素，如果加上等于号，就会执行else下面的语句，把该元素填进结果数组。

出自：代码随想录