程序在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 好听的元旦简短祝福语