977. Squares of a Sorted Array

題目：

Given an integer array nums sorted in non-decreasing order, return an array of the squares of each number sorted in non-decreasing order.

 

Example 1:

Input: nums = [-4,-1,0,3,10]
Output: [0,1,9,16,100]
Explanation: After squaring, the array becomes [16,1,0,9,100].
After sorting, it becomes [0,1,9,16,100].

Example 2:

Input: nums = [-7,-3,2,3,11]
Output: [4,9,9,49,121]

 

Constraints:

• 1 <= nums.length <= 10^4
• -104 <= nums[i] <= 10^4
• nums is sorted in non-decreasing order.

 

 

思路：

這題資料量較小，只有10的4次方，因此直接在原陣列上進行平方然後再sort也可以通過。這裡說一個雙指標one pass的方法。先建立大小相等的空陣列來記錄答案，然後左右指標分別指向原陣列兩端，誰的絕對值/平方大就選誰，從空陣列的後面開始放，一路向前即可。

 

 

程式碼：

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n=nums.size();
        vector<int> res(n);
        int l=0, r=n-1;
        for(int i=n-1;i>=0;i--)
        {
            if(abs(nums[l])>abs(nums[r]))
            {
                res[i]=nums[l]*nums[l];
                l++;
            }
            else
            {
                res[i]=nums[r]*nums[r];
                r--;
            }
        }
        return res;
    }
};