"""
题目:输入一个二维矩阵,如何计算给定左上角坐标和右下角坐标的子矩阵之和?
例如输入下面的二维矩阵,以及左上角坐标为(2,1)和右下角坐标为(4,3)的子矩阵,该函数输出8
解答思路:书上的答案还是繁琐,可能是因为使用Java的原因吧,对于python而言,将数组的每行视为一个子列表,
然后使用列表切片功能,对每个子列表切片求和即可
"""
t = [
[3, 0, 1, 4, 2],
[5, 6, 3, 2, 1],
[1, 2, 0, 1, 5],
[4, 1, 0, 1, 7],
[1, 0, 3, 0, 5]
]
location_start = (2, 1)
location_end = (4, 3)
def numpy_summary(t, start, end):
start_x, start_y = start
end_x, end_y = end
summary = 0
for k in range(end_x-start_x+1):
sub_summary = sum(t[start_x+k][start_y:end_y+1])
summary += sub_summary
return summary
summary = numpy_summary(t, location_start, location_end)
print(summary)
《剑指offer》面试题13:二维子矩阵的数字之和
最新推荐文章于 2023-12-24 18:05:30 发布