Python 中的康威生命游戏 - 竞争性编程 - 如何优化

Question

我正在解决 生命游戏问题.com/ieeextreme-practice/task/d12a7784af1a3d3f6d88601c81a4bb81/" rel="nofollow noreferrer">csacademy 并且我无法在更大的投入上击败时间。对优化代码有帮助吗？

我尝试更改一些内容，例如使用 np.array() 而不是 list，并且不将原始输入转换为 1 和 0（原始输入是“*”和“-”，需要以这种方式打印）。

from copy import deepcopy
import numpy as np
aliveToDead = {0, 1, 4, 5, 6, 7, 8}
deadToAlive = {3}


def countNeighbors(M, n, m, i, j):
    s = M[i, (j + 1) % m] + M[i, j - 1] + M[(i + 1) % n, j] + M[i - 1, j]
    s += M[i - 1, j - 1] + M[(i + 1) % n, (j + 1) % m] + M[i - 1, (j + 1) % m] + M[(i + 1) % n, j - 1]
    return s


def gameOfLife(mat, n, m, C):
    cells = deepcopy(mat)
    for c in range(C):
        for i in range(n):
            for j in range(m):
                neighbors = countNeighbors(mat, n, m, i, j)
                if mat[i, j] == 1 and neighbors in aliveToDead:
                    cells[i, j] = 0
                elif mat[i, j] == 0 and neighbors in deadToAlive:
                    cells[i, j] = 1
        mat = deepcopy(cells)
    return mat


def buildList(n):
    return np.array([[0 if x == '-' else 1 for x in input()] for i in range(n)])


def printResult(mat):
    mat = mat.astype(str)
    mat[mat == "1"] = '*'
    mat[mat == "0"] = '-'
    for row in mat:
        print(*row, sep="")


def main():
    n, m, c = map(int, input().split())
    mat = buildList(n)
    result = gameOfLife(mat, n,  m, c)
    printResult(result)


if __name__ == "__main__":
    main()

Answer 1

这个数学解决方案似乎有助于通过更多测试，但仍有一些失败。

def gameOfLife(mat, n, m, C):
    cells = deepcopy(mat)
    loop = n*m*4*3*5
    while loop % 16:
        loop *= 2
    num_iter = C % loop
    for c in range(num_iter):
        for i in range(n):
            for j in range(m):
                neighbors = countNeighbors(mat, n, m, i, j)
                if mat[i, j] == 1:
                    cells[i, j] = aliveToDead[neighbors]
                else:
                    if neighbors==3:
                        cells[i, j] = 1
                    else:
                        cells[i, j] = 0
        mat,cells = cells,mat
    return mat