leetcode 977. Squares of a Sorted Array

题目描述

双指针法一

用right表示原数组中负数和非负数的分界线。

nums[0,right-1]的是负数，nums[right,nums.size()-1]是非负数。

然后用合并两个有序数组的方法。合并即可。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> res;
        int len = nums.size();
        res.reserve(len);
        int right = -1;
        for(int i =0;i <len;i++){
            if(right == -1 && nums[i] >=0)
                right = i;
            nums[i] = nums[i]*nums[i];
        }
        int left = right-1;
        if(right == -1){
            right = len;
            left = len -1;
        }

        while(left >= 0 && right < len){
            if(nums[left]<=nums[right]){
                res.push_back(nums[left--]);
            }else{
                res.push_back(nums[right++]);
            }
        }
        while(left>=0){
            res.push_back(nums[left--]);
        }
        while(right<len){
            res.push_back(nums[right++]);
        }
        return res;
    }
};

双指针法二

容易发现原数组中元素的绝对值从两端向中间减小。因此可以逆序确定res数组。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int len = nums.size();
        vector<int> res(len,0);
        int left = 0;
        int right = len -1;
        int res_pos = len - 1;
        while(left <= right){
            if(abs(nums[left]) > abs(nums[right])){
                res[res_pos--] = nums[left]*nums[left];
                left++;
            }else{
                res[res_pos--] = nums[right]*nums[right];
                right--;
            }
        }
        return res;
    }
};