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]);
}
}
}