Leetcode977:有序数组的平方
. - 力扣(LeetCode). - 备战技术面试?力扣提供海量技术面试资源,帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/squares-of-a-sorted-array/description/
解题思路
输入是非递减顺序的整数数组,说明数组中平方最大的数不是第一个就是最后一个。
例如,[-4,-1,0,3,10],数组中最大的正整数是最后一个,最小的负整数是第一个,它们的平方都有可能是最大的。使用双指针比较第一个数和最后一个数的平方大小,可以较快地解决问题。
可运行的代码
public class Solution {public int[] sortedSquares(int[] nums) {int k = nums.length;int[] res = new int[k];int left = 0, right = k - 1;while (left <= right) {if (nums[left] * nums[left] > nums[right] * nums[right]) {res[--k] = nums[left] * nums[left];left++;} else {res[--k] = nums[right] * nums[right];right--;}}return res;}public static void main(String[] args) {int[] nums = {-4, -1, 0, 3, 10};Solution solution = new Solution();int[] res = solution.sortedSquares(nums);for (int i = 0; i < res.length; i++) {System.out.println(res[i]);}}
}