给你一个 m * n 的矩阵,矩阵中的数字 各不相同 。请你按 任意 顺序返回矩阵中的所有幸运数。
幸运数是指矩阵中满足同时下列两个条件的元素:
在同一行的所有元素中最小
在同一列的所有元素中最大
输入:matrix = [[3,7,8],[9,11,13],[15,16,17]]
输出:[15]
解释:15 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
输入:matrix = [[1,10,4,2],[9,3,8,7],[15,16,17,12]]
输出:[12]
解释:12 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
解1
# 遍历矩阵,判断matrix[i][j] 是否大于等于同一列的最大值,是否
# 小于等于同一行的最小值
class Solution:
def luckyNumbers (self, matrix: List[List[int]]) -> List[int]:
ans = []
for row in matrix:
for j,x in enumerate(row):
if max(r[j] for r in matrix) <= x <= min(row):
ans.append(x)
return ans
获取同一列的最大值: 当j=0,r是第一行时,r[j] 是matrix[0][0]
r是第二行时,r[j] 是 matrix[1][0]
同一行的最小值: min(row)
解2
x, y, ans = [], [], []
# 每一行的最小值存入 x
for i in matrix:
x.append(min(i))
# zip(*matrix)将数组转置,所以遍历的 i 就是每一列
for i in zip(*matrix):
# 将每一列的最大值存入 y
y.append(max(i))
for i in x:
if i in y:
ans.append(i)
print(ans)
python zip()函数
Python 3.x 中为了减少内存,zip() 返回的是一个对象。如需展示列表,需手动 list() 转换。
zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由
这些元组组成的列表。
实例
num1 = [1, 2, 3, 4, 5]
a = zip(num1)
print(list(a))
# [(1,), (2,), (3,), (4,), (5,)]
num2 = [6, 7, 8, 9, 10]
a = zip(num1,num2)
print(list(a))
# [(1, 6), (2, 7), (3, 8), (4, 9), (5, 10)]
num3 = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
num4 = [[2, 2, 2], [3, 3, 3], [4, 4, 4]]
print(list(zip(num3,num4)))
# [([1, 2, 3], [2, 2, 2]), ([4, 5, 6], [3, 3, 3]), ([7, 8, 9], [4, 4, 4])]
print(list(zip(*num3)))
# [(1, 4, 7), (2, 5, 8), (3, 6, 9)]