《剑指offer》面试题13：二维子矩阵的数字之和

"""
题目：输入一个二维矩阵，如何计算给定左上角坐标和右下角坐标的子矩阵之和？
     例如输入下面的二维矩阵，以及左上角坐标为（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)