Leetcode 498. 对角线遍历

思路找规律

1. 设x,y为数组下标，x+y为偶数时，向右上遍历。x+y为奇数时，向左下遍历。

2. 在向左下遍历时先考虑边界1 当前在最后一行，则向右移动一格 y+1；再考虑边界 2 当前在第一列，向下移动一格 x+1; 其他情况 x+1,y-1

3. 在向右上遍历时先考虑边界1 当前在最后一列，向下移动一格 x+1; 再考虑边界 2 当前在第一行时，向右移动一格 y+1; 其他情况 y+1, x-1

func findDiagonalOrder(mat [][]int) []int {
    m := len(mat)
    n := len(mat[0])
    elementCount := m * n
    res := make([]int, 0)
    x, y := 0, 0
    for i := 0; i < elementCount; i++ {
        res = append(res, mat[x][y])
        if (x+y)%2 == 0 {
            if y == n-1 {
                x++
            } else if x == 0 {
                y++
            } else {
                x--
                y++
            }
        } else {
            if x == m-1 {
                y++
            } else if y == 0 {
                x++
            } else {
                x++
                y--
            }
        }
    }
    return res
}

__EOF__

本文作者：SoutherLea
本文链接：https://www.cnblogs.com/lizhengnan/p/16251945.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！