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]
题解:
可以采用双指针来解决,我们只需要维护一个存放平方后的新数组,初始化左右指针和新数组的下标为0,nums.length-1,nums.length-1,因为数组里的数可正可负,所以我们可以知道,最大值肯定在数组的最左端或者最右端产生,所以我们只需要比较左右指针所指位置的数的平方的大小,将其中大的数存入到新数组中,这样就可以得到从小到大排序的新数组,如图所示
Python:
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
left,right,pose = 0,len(nums)-1,len(nums)-1
L = [0]*len(nums)
while left <= right:
if nums[left]*nums[left] > nums[right]*nums[right]:
L[pose] = nums[left]*nums[left]
left += 1
else:
L[pose] = nums[right]*nums[right]
right -= 1
pose -= 1
return L
C++:
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
int n = nums.size();
vector<int> arr(n);
int left = 0;
int right = n-1;
int pose = n-1;
while (left <= right)
{
if (nums[left]*nums[left] > nums[right]*nums[right])
{
arr[pose] = nums[left] * nums[left];
left++;
}else{
arr[pose] = nums[right] * nums[right];
right--;
}
pose--;
}
return arr;
}
};
Java:
class Solution {
public int[] sortedSquares(int[] nums) {
int n = nums.length;
int[] arr = new int[n];
for (int left = 0, right=n-1, pose=n-1; left <= right; pose--)
{
if(nums[left]*nums[left] > nums[right]*nums[right])
{
arr[pose] = nums[left]*nums[left];
left ++;
}else{
arr[pose] = nums[right]*nums[right];
right--;
}
}
return arr;
}
}