LeetCode 筆記 - 977. Squares of a Sorted Array

題目在此 977. Squares of a Sorted Array

給一個 Sorting 過的數列,請將每個數字平方之後,再做一次排序

解題思維

直接做可以過

要加速可以在做平方的時候把負數都挑起來,然後針對只剩下自然數的數列做 merge
這樣可以加速一些 Sorting 的過程

程式碼

1
2
3
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
return sorted(list(map(lambda x: x ** 2, nums)))

相關文章