Python - 使用递归遍历矩阵 [重复]

Question

这个问题在这里已经有了答案：

为什么我的递归函数返回 None？ (4 个回答)

递归函数在 Python 中不返回任何内容 [重复] （2 个回答）

18 小时前关闭。

假设我有一个这样的矩阵：

array = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

我想制作一种方法，它基本上遍历矩阵并打印出如下整数：

1,2,3,4,5,6,7,8,9

但我想递归地执行此操作，以便获得更多经验。 我目前有一个帮手的这个方法：

def flatten_array(matrix):
    row = len(matrix)
    col = len(matrix[0])
    return str(helper(row, col, 0, 0, matrix))


def helper(current_row, current_col, i, j, matrix):
    if i >= current_row:
        return 1
    elif j >= current_col:
        helper(current_row, current_col, i + 1, 0, matrix)
    else:
        print(matrix[i][j], end=',')
        helper(current_row, current_col, i, j + 1, matrix)

它几乎打印了我想要的东西，它打印了这个：

1,2,3,4,5,6,7,8,9,None

如何在没有额外逗号和 None 而不是打印的情况下返回它？

Answer 1

您只需要检查是否已到达最后一个元素，然后就不要打印逗号。

def helper(current_row, current_col, i, j, matrix):
    if i >= current_row:
        return 1
    elif j >= current_col:
        helper(current_row, current_col, i + 1, 0, matrix)
    else:
        if i == len(matrix)-1 and j == len(matrix[0])-1:
            print(matrix[i][j])
        else:
            print(matrix[i][j], end=',')
        helper(current_row, current_col, i, j + 1, matrix)

我试图尽可能接近你的代码。 如果您需要进一步解释，请发表评论。