其他分享
首页 > 其他分享> > Leetcode 498. 对角线遍历

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
}

  

标签:遍历,mat,++,res,else,int,498,Leetcode
来源: https://www.cnblogs.com/lizhengnan/p/16251945.html