动物牛国度的农场上有一片田地,农场主给每个农场工人分配了一块 m x n 的矩阵 mat,每个元素表示该位置的作物数量。现在,请你编写一个函数,按照对角线遍历的顺序,将矩阵中的所有元素放入一个数组中并返回。 给定一个大小为 m x n 的矩阵 mat,请你实现函数 vector diagonalTraverse(vector& mat),返回按照对角线遍历顺序得到的元素数组。
示例1

输入

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

输出

[1,2,4,7,5,3,6,8,9]
示例2

输入

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

输出

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

备注:
m 和 n 分别是矩阵 mat 的行数和列数。1 1 0
加载中...