leetcode-977.有序数组的平方

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

示例 1：

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

输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

提示：

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按 非递减顺序 排序

进阶：

请你设计时间复杂度为 O(n) 的算法解决本问题

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/squares-of-a-sorted-array

#include<iostream>
#include<vector>
using namespace std;
/*思路：数组内元素平方后其实可以看做已经排序好，左右两端向中间靠拢是越来越小的，只需要区分左边有负数时左右两边谁大。用双指针从两端向中间扫描，另用一个新数组从大到小存放数据,时间空间复杂度皆O(n)*/
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> _Squares(nums.size(), 0);//新建初始化一个数组
        int i = 0;
        size_t j = nums.size() - 1;
        size_t k = j;
        while (i <= j)
        {
            if (nums[i] * nums[i] < nums[j] * nums[j])
            {
                _Squares[k--] = nums[j] * nums[j];
                --j;
            }
            else
            {
                _Squares[k--] = nums[i] * nums[i];
                ++i;
            }
        }
        return _Squares;
    }
};
//test
int main()
{   
    vector<int> _nums;
    int n, m;
    cin >> n;
    for (int i = 0; i < n; ++i)
    {
        cin >> m;
        _nums.push_back(m);
    }
    Solution s1;
    for (auto a : s1.sortedSquares(_nums))
        cout << a << endl;
}