题目:
给你一个按 非递减顺序 排序的整数数组 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]
C语言:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
void sort(int a[], int m)
{
int i=0,j=0,tmp=0;
for (i = 0; i<m - 1; i++)
for (j = 0; j < m - i - 1; j++)
if (a[j]>a[j + 1])
{
tmp = a[j + 1];
a[j + 1] = a[j];
a[j] = tmp;
}
}
int* sortedSquares(int* nums, int numsSize, int* returnSize)
{
*returnSize = numsSize;
for(int i=0;i<numsSize;i++)
nums[i] *= nums[i];
sort(nums,numsSize);
return nums;
}
执行用时1516ms,内存消耗19.2MB。
Python3:
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
return sorted([i**2 for i in nums])
执行用时56ms,内存消耗16.4MB。