程序在Python中查找下一个单元矩阵状态的下一个状态？

假设我们有一个二维二进制矩阵，其中1表示活细胞，0表示死细胞。一个单元的邻居是它的直接水平，垂直和对角单元。我们必须使用这些规则找到矩阵的下一个状态

• 任何有两个或三个活着邻居的活细胞都可以生活。

• 具有三个活着邻居的任何死细胞都将成为活细胞。

• 所有其他细胞死亡。

所以，如果输入像

那么输出将是

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

• n：=矩阵的行大小，m：=矩阵的列大小

• res：=大小为nxm的矩阵，并用0填充

• 对于0到n范围内的i，执行

• s：= 0

• 如果matrix [i，j]等于0，则

• 除此以外，

• 或h在j-1至j + 1的范围内，执行

• s：= s +矩阵[k，h]

• 如果0 <= k <n和0 <= h <m

• 对于范围i-1至i + 1的k

• res [i，j]：= [0，1，当s与3相同时为真]

• res [i，j]：= 1

• 对于范围j-1至j + 1中的h

• s：= s +矩阵[k，h]

• 如果0 <= k <n和0 <= h <m

• 对于范围i-1至i + 1的k

• 如果s是3或4，则

• 对于0到m范围内的j，执行

• 返回资源

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

示例

class Solution:
   def solve(self, matrix):
      n, m = len(matrix), len(matrix[0])
      res = [[0 for j in range(m)] for i in range(n)]
      for i in range(n):
         for j in range(m):
            s = 0
            if matrix[i][j] == 0:
               for k in range(i - 1, i + 2):
                  for h in range(j - 1, j + 2):
                     if 0 <= k < n and 0 <= h < m:
                        s += matrix[k][h]
               res[i][j] = [0, 1][s == 3]
            else:
               for k in range(i - 1, i + 2):
                  for h in range(j - 1, j + 2):
                     if 0 <= k < n and 0 <= h < m:
                        s += matrix[k][h]
               if s in [3, 4]:
                  res[i][j] = 1
      return res

ob = Solution()matrix = [
   [1, 1, 0, 0],
   [0, 1, 0, 0],
   [0, 1, 0, 1],
   [1, 1, 0, 1]
]

print(ob.solve(matrix))

输入值

[[1, 1, 0, 0],
 [0, 1, 0, 0],
 [0, 1, 0, 1],
 [1, 1, 0, 1] ]

输出结果

[[1, 1, 0, 0],
 [0, 1, 0, 0],
 [0, 1, 0, 0],
 [1, 1, 0, 0]]