Leetcode 498. 对角线遍历

498. 对角线遍历 - 力扣（LeetCode）

 

 

 

思路 找规律

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
}