LeetCode数组篇【有序数组的平方】

977.有序数组的平方

力扣题目链接(opens new window)

给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序

正解

采用双指针的方法，由于新的数组值都是平方的结果，所以最大值到最小值就对应着旧数组两边到中间，所以可以设置一个左右指针，依次比较平方的大小并进行记录，直至两指针相交找的最小值，历遍结束。

class Solution {
        public int[] sortedSquares(int[] nums) {
        int left = 0;
        int right = nums.length - 1;
        int nnums[] = new int[nums.length];
        int index=nums.length-1;
        while (left <=right) {
            if(Math.pow((nums[left]),2)>=Math.pow((nums[right]),2)){
                nnums[index--]= (int) Math.pow((nums[left++]),2);
            }else {
                nnums[index--]= (int) Math.pow((nums[right--]),2);
            }
        }
        return nnums;
    }
}

（值得提醒的是这里的Math.pow可以换为a*a的形式，这样可以省去Math.pow方法的调用，减少运行的时间，不过这种方式相对来说更节约内存，可以选择使用）

python版：

class Solution(object):
    def sortedSquares(self, nums):
        left=0
        right=len(nums)-1
        nnums=[None] * len(nums)
        index=len(nums)-1
        while left<=right:
            if nums[left]**2>=nums[right]**2:
                nnums[index]=nums[left]**2
                left+=1
                index-=1
            else:
                nnums[index]=nums[right]**2
                right-=1
                index-=1
        return nnums