遍历二维数组的对角线可以通过以下代码实现:
def traverse_diagonal(arr):
rows = len(arr)
cols = len(arr[0])
# 遍历左上角到右下角的对角线
for i in range(min(rows, cols)):
print(arr[i][i])
# 遍历右上角到左下角的对角线
for i in range(min(rows, cols)):
print(arr[i][cols-1-i])
这里假设输入的二维数组是一个矩阵,即每一行的元素个数相等。
你可以使用traverse_diagonal函数来遍历任意二维数组的对角线。例如,如果有以下二维数组:
arr = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
调用traverse_diagonal(arr)将输出:
1
5
9
3
7
这样就完成了对二维数组的对角线的遍历。
下一篇:遍历二维数组的性能问题