python leetcode刷题 (6):977.有序数组的平方

题目描述:

给定一个按非递减顺序排序的整数数组 A,返回每个数字的平方组成的新数组,要求也按非递减顺序排序。

示例 1:

输入:[-4,-1,0,3,10]
输出:[0,1,9,16,100]
示例 2:

输入:[-7,-3,2,3,11]
输出:[4,9,9,49,121]

提示:

1 <= A.length <= 10000
-10000 <= A[i] <= 10000
A 已按非递减顺序排序。

解题过程:

代码如下:

class Solution:
    def sortedSquares(self, A: List[int]) -> List[int]:
        B=[]
        for a in A:
            b=pow(a,2)
            B.append(b)
        B.sort()
        return B

另有采用列表表达式的一行解法:

class Solution:
    def sortedSquares(self, A: List[int]) -> List[int]:
        return sorted([x**2 for x in A])

总结:

  1. pow()是平方,sqrt()是开平方
  2. 列表变成列表的题目可以使用列表表达式
  3. 列表升序排序可以使用sort()和sorted(),sort()排序完返回原有列表,而sorted排序完返回新列表

参考文献:

https://www.cnblogs.com/whaben/p/6495702.html python 列表排序方法sort、sorted技巧篇

猜你喜欢

转载自blog.csdn.net/weixin_42785446/article/details/89071427