leetcode498 对角线遍历

题目

给你一个大小为 m x n 的矩阵 mat ，请以对角线遍历的顺序，用一个数组返回这个矩阵中的所有元素。

示例

输入：mat = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,4,7,5,3,6,8,9]

解析

本题目主要考察的就是模拟法，首先可以计算得出，对角线的个数为m+n-1，在此基础上对于对角线进行遍历（从0开始），则第偶数条为从左下到右上，第奇数条为从右上到左下，根据奇数偶数来确定走什么遍历逻辑。
在遍历的过程中，由于边界值并不固定，每次处理的时候需要判断起点的横纵坐标值：

func findDiagonalOrder(mat [][]int) (ans []int) {m := len(mat)n := len(mat[0])var x, y intfor i := 0; i < m+n-1; i++ { // 对角线的个数if i%2 == 0 {// 偶数，从左下到右上if i < m {x, y = i, 0} else {x, y = m-1, i-m+1}for x >= 0 && y < n {ans = append(ans, mat[x][y])x--y++}} else {// 奇数，从右上到左下if i < n {x, y = 0, i} else {x, y = i-n+1, n-1}for x < m && y >= 0 {ans = append(ans, mat[x][y])x++y--}}}return
}