字节跳动面试题:有序数组的平方
描述
给定一个按非递减顺序排序的整数数组 A,返回每个数字的平方组成的新数组,要求也按非递减顺序排序。
- 1 <= A.length <= 10000
- -10000 <= A[i] <= 10000
- A 已按非递减顺序排序。
样例1
输入:
[-4,-1,0,3,10]
输出:
[0,1,9,16,100]
示例2
输入:
[-7,-3,2,3,11]
输出:
[4,9,9,49,121]
算法: sort排序
解题思路
直接原地平方,然后进行排序
复杂度分析
时间复杂度:O(nlogn)
排序需要nlogn的时间复杂度,n为数组长度。
空间复杂度:O(1)
不需要开辟新的额外空间,n为数组长度。
源代码
class Solution:
"""
@param A: The array A.
@return: The array of the squares.
"""
def SquareArray(self, A):
for i in range(len(A)):
A[i] = A[i] * A[i]
A.sort()
return A