2019-12-29-LeetCode【977. 有序数组的平方】

977. 有序数组的平方

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

示例 1：

输入：[-4,-1,0,3,10]
输出：[0,1,9,16,100]
示例 2：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

提示：

1. 1 <= A.length <= 10000
2. -10000 <= A[i] <= 10000
3. A 已按非递减顺序排序。

思路

充分利用题目条件，平方大的肯定在首尾两边。所以定义两个指针，分别指向头和尾，将两个指针指向的元素大的存放到新的数组中，然后让该指针向中间移动，再比较两个指针指向元素的平方的大小，直到遍历所有的元素。

代码

class Solution {
public:
    //排序，错误。。。
    // vector<int> sortedSquares(vector<int>& A) {

        
    //     int len = A.size();

    //     for(auto i=0;i<len;i++)
    //     {
    //         A[i]=A[i]*A[i];
    //     }
    //     sort(A.begin(),A.end());
    //     return A;
        
    // }


    //双指针
     vector<int> sortedSquares(vector<int>& A) {
         int len=A.size();
         vector<int> ans(len);
         int left=0,right=len-1,k=len-1;
         while(k>=0)
         {
             int a=A[left]*A[left];
             int b=A[right]*A[right];
             if(a>b)
             {
                 ans[k]=a;
                 left++;
             }
             else
             {
                 ans[k]=b;
                 right--;
             }
             k--;
         }

        return ans;

     }
};