题目链接:977. 有序数组的平方 - 力扣(LeetCode)
给你一个按 非递减顺序 排序的整数数组 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
已按 非递减顺序 排序
暴力排序:
java:
class Solution {public int[] sortedSquares(int[] nums) {for(int i=0;i<nums.length;i++){nums[i]=nums[i]*=nums[i];}Arrays.sort(nums);return nums;}
}
C++:
class Solution {
public:vector<int> sortedSquares(vector<int>& nums) {for(int i=0;i<nums.size();i++){nums[i]=nums[i]*=nums[i];}sort(nums.begin(),nums.end());return nums;}
};
双指针法:
由于所给的nums数组是排好序的,所以最大值会出现在数组两端,使用双指针从两端向中间遍历,从尾到头(数值从大到小)更新数组
class Solution {public int[] sortedSquares(int[] nums) {//双指针法int left = 0;int right = nums.length - 1;int k = nums.length - 1;int[] ans = new int[nums.length];while (left <= right) {if (nums[left] * nums[left] >= nums[right] * nums[right]) {ans[k] = nums[left] * nums[left];k--;left++;} else {ans[k] = nums[right] * nums[right];k--;right--;}}return ans;}
}