代码随想录刷题——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) 的算法解决本问题

解题思路

在没看到进阶要求之前，我就想直接全部平方然后暴力求解了哈哈，一行就可以解决：

public int[] sortedSquares(int[] nums) {
    return Arrays.stream(nums).map(item -> item = item * item).sorted().toArray();
}

不过效率是比较低的。

这道题比较高效的做法就是使用双指针，因为数组本身是有序的，分别在首尾置一个指针，声明一个数组用于存储结果，比较指针所指向的元素的平方和的大小，较大的存储到结果数组的后面，然后移动相应的指针。代码如下：

public int[] sortedSquares(int[] nums) {
    int left = 0;
    int right = nums.length - 1;
    int[] res = new int[nums.length];
    int index = nums.length - 1;
    while(left <= right){
        int powleft = nums[left] * nums[left];
        int powright = nums[right] * nums[right];
        if(powleft > powright){
            res[index--] = powleft;
            left++;
        } else {
            res[index--] = powright;
            right--;
        }
    }
    return res;
}

博客原文地址

代码随想录刷题——977. 有序数组的平方