力扣977.有序数组的平方
题目链接: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;
}
}