提交 bd73b753 编写于 作者: 檀越@新空间's avatar 檀越@新空间 🐭

fix:矩阵中的和

上级 968a311c
"""
矩阵中的和
"""
from typing import List
class Solution:
def matrixSum(self, nums: List[List[int]]) -> int:
"""
行转列 先遍历矩阵,保存index-max的关系
:param nums:
:return:
"""
for row in nums:
row.sort()
res = 0
dict1 = {}
for i in range(len(nums)):
for j in range(len(nums[0])):
if j in dict1:
dict1[j] = max(dict1[j], nums[i][j])
else:
dict1[j] = nums[i][j]
for key in dict1:
res += dict1[key]
return res
if __name__ == '__main__':
result = Solution().matrixSum([[1, 8, 16, 15, 12, 9, 15, 11, 18, 6, 16, 4, 9, 4],
[3, 19, 8, 17, 19, 4, 9, 3, 2, 10, 15, 17, 3, 11],
[13, 10, 19, 20, 6, 17, 15, 14, 16, 8, 1, 17, 0, 2],
[12, 20, 0, 19, 15, 10, 7, 10, 2, 6, 18, 7, 7, 4],
[17, 14, 2, 2, 10, 16, 15, 3, 9, 17, 9, 3, 17, 10],
[17, 6, 19, 17, 18, 9, 14, 2, 19, 12, 10, 18, 7, 9],
[5, 6, 5, 1, 19, 8, 15, 2, 2, 4, 4, 1, 2, 17],
[12, 16, 8, 16, 7, 6, 18, 13, 18, 8, 14, 15, 20, 11],
[2, 10, 19, 3, 15, 18, 20, 10, 6, 7, 0, 8, 3, 7],
[11, 5, 10, 13, 1, 3, 4, 7, 1, 18, 20, 17, 19, 2],
[0, 3, 20, 6, 19, 18, 3, 12, 2, 11, 3, 1, 19, 0],
[6, 5, 3, 15, 6, 1, 0, 17, 13, 19, 3, 8, 2, 7],
[2, 20, 9, 11, 13, 5, 1, 16, 14, 1, 19, 3, 12, 6]])
print(result)
"""
矩阵中的和
"""
from typing import List
import numpy as np
class Solution:
def matrixSum(self, nums: List[List[int]]) -> int:
"""
axis=0表示列
axis=1表示行
:param nums:
:return:
"""
for i in nums:
i[:] = sorted(i)
return int(sum(np.max(np.array(nums), axis=0)))
if __name__ == '__main__':
result = Solution().matrixSum([[1, 8, 16, 15, 12, 9, 15, 11, 18, 6, 16, 4, 9, 4],
[3, 19, 8, 17, 19, 4, 9, 3, 2, 10, 15, 17, 3, 11],
[13, 10, 19, 20, 6, 17, 15, 14, 16, 8, 1, 17, 0, 2],
[12, 20, 0, 19, 15, 10, 7, 10, 2, 6, 18, 7, 7, 4],
[17, 14, 2, 2, 10, 16, 15, 3, 9, 17, 9, 3, 17, 10],
[17, 6, 19, 17, 18, 9, 14, 2, 19, 12, 10, 18, 7, 9],
[5, 6, 5, 1, 19, 8, 15, 2, 2, 4, 4, 1, 2, 17],
[12, 16, 8, 16, 7, 6, 18, 13, 18, 8, 14, 15, 20, 11],
[2, 10, 19, 3, 15, 18, 20, 10, 6, 7, 0, 8, 3, 7],
[11, 5, 10, 13, 1, 3, 4, 7, 1, 18, 20, 17, 19, 2],
[0, 3, 20, 6, 19, 18, 3, 12, 2, 11, 3, 1, 19, 0],
[6, 5, 3, 15, 6, 1, 0, 17, 13, 19, 3, 8, 2, 7],
[2, 20, 9, 11, 13, 5, 1, 16, 14, 1, 19, 3, 12, 6]])
print(result)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册