LeetCode-766

题目

结果

代码

func isToeplitzMatrix(matrix [][]int) bool {
	// only one element
	if len(matrix) == 1 && len(matrix[0]) == 1 {
		return true
	}
	// it's not a matrix but a single row or column
	if len(matrix) < 2 || len(matrix[0]) < 2 {
		return true
	}
	for i := 0; i < len(matrix) || i < len(matrix[0]); i++ {
		if i < len(matrix[0]) {
			v1 := matrix[0][i]
			for row, col := 1, i+1; col < len(matrix[0]) && row < len(matrix); {
				if matrix[row][col] != v1 {
					return false
				}
				row++
				col++
			}
		}

		if i < len(matrix) {
			v2 := matrix[i][0]
			for row, col := i+1, 1; row < len(matrix) && col < len(matrix[0]); {
				if matrix[row][col] != v2 {
					return false
				}
				row++
				col++
			}
		}
	}
	return true
}