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 }