给定一个从小到大排序的整数数组A,然后将每个整数的平方和从小到大排序。
Example 1:
Input: [-4,-1,0,3,10]
Output: [0,1,9,16,100]
Example 2:
Input: [-7,-3,2,3,11]
Output: [4,9,9,49,121]
直接求和每个数的平方和,然后再从小到大排序
class Solution:
def sortedSquares(self, A: 'List[int]') -> 'List[int]':
return sorted([x**2 for x in A])
假设有两个指针,分别从头和从尾向中间移动,然后比较左右两个值的绝对值大小,绝对值大的将平方和添加到最左边,一直到两个指针相遇。
class Solution:
def sortedSquares(self, A):
answer = [0] * len(A)
l, r = 0, len(A) - 1
while l <= r:
left, right = abs(A[l]), abs(A[r])
if left > right:
answer[r - l] = left * left
l += 1
else:
answer[r - l] = right * right
r -= 1
return answer