程序在Python中查找包含后面行和列总和的行和列的矩阵
假设我们有一个给定的矩阵,我们必须找到一个新的矩阵res,其维度与给定的矩阵相同,其中res[i,j]中的每个元素=矩阵[r,c]中每个元素的总和r≤i,且c≤j。
所以,如果输入是这样的
那么输出将是
示例
让我们看下面的实现来更好地理解
def solve(matrix):
if not matrix:
return matrix
R, C = len(matrix), len(matrix[0])
for r in range(1, R):
for c in range(C):
matrix[r][c] += matrix[r - 1][c]
for r in range(R):
for c in range(1, C):
matrix[r][c] += matrix[r][c - 1]
return matrix
matrix = [
[8, 2],
[7, 4]
]
print(solve(matrix))输入
[[8, 2],[7, 4]]输出结果
[[8, 10], [15, 21]]
热门推荐
10 新人红包祝福语简短精辟
11 幼师给毕业祝福语简短
12 生日高考祝福语妹妹简短
13 下飞机祝福语简短英文
14 元宵发客户祝福语简短
15 爸妈金婚文案祝福语简短
16 新郎朋友红包祝福语简短
17 虎年五一祝福语大全简短
18 出阁宴祝福语姐姐简短