在 Python 中已知矩阵每行和每列的最大元素，求原矩阵

假设我们有两个数组 A 和 B，大小分别为 N 和 M，我们还有一个 N x M 的二元矩阵，其中 1 表示原矩阵中存在正整数，0 表示该位置在原矩阵中为 0。我们必须生成原矩阵，使得 A[i] 表示第 i 行的最大元素，B[j] 表示第 j 列的最大元素。

因此，如果输入类似于 A = [4, 2, 3]，B = [3, 1, 0, 0, 4, 0, 5] 矩阵，则输出将是矩阵

为了解决这个问题，我们将遵循以下步骤：

  • N := A 的大小

  • M := B 的大小

  • 对于 i 的范围为 0 到 N，执行：

    • 对于 j 的范围为 0 到 M，执行：

      • 如果 mat[i, j] 等于 1，则

        • 显示 A[i] 和 B[j] 的最小值

      • 否则，

        • 打印换行符

示例

让我们看看下面的实现，以便更好地理解：

在线演示

def print_original_mat(A, B, mat) :
   N = len(A)
   M = len(B)
   for i in range(N) :
      for j in range(M) :
         if (mat[i][j] == 1) :
            print(min(A[i], B[j]), end = " ")
         else :
            print(0, end = " ")
      print()
A = [4, 2, 3]
B = [3, 1, 0, 0, 4, 0, 5]
mat = [
   [1, 0, 0, 0, 1, 0, 1],
   [0, 0, 1, 0, 0, 1, 1],
   [1, 1, 0, 1, 1, 0, 0]]
print_original_mat(A, B, mat);

输入

[4, 2, 3],
[3, 1, 0, 0, 4, 0, 5],
[[1, 0, 0, 0, 1, 0, 1],
[0, 0, 1, 0, 0, 1, 1],
[1, 1, 0, 1, 1, 0, 0]]

输出

3 0 0 0 4 0 4
0 0 0 0 0 0 2
3 1 0 0 3 0 0

更新于：2020年8月27日

178 次浏览

启动你的职业生涯

完成课程获得认证

开始学习
广告