LeetCode-766 托普利茨矩阵

如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素,那么这个矩阵是托普利茨矩阵

给定一个 M x N 的矩阵,当且仅当它是托普利茨矩阵时返回 True

示例 1:

输入: 
matrix = [
  [1,2,3,4],
  [5,1,2,3],
  [9,5,1,2]
]
输出: True
解释:
在上述矩阵中, 其对角线为:
"[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。
各条对角线上的所有元素均相同, 因此答案是True。

示例 2:

输入:
matrix = [
  [1,2],
  [2,2]
]
输出: False
解释: 
对角线"[1, 2]"上的元素不同。

说明:

  1.  matrix 是一个包含整数的二维数组。
  2. matrix 的行数和列数均在 [1, 20]范围内。
  3. matrix[i][j] 包含的整数在 [0, 99]范围内。

思路

注意边界,遍历判断就行

func isToeplitzMatrix(matrix [][]int) bool {
	M := len(matrix)
	N := len(matrix[0])
	for i:=2;i<=M;i++{
		var x,y int
		y = M - i
		for j:=0;j<i-1 && j<N-1;j++{
			if matrix[y][x] != matrix[y+1][x+1]{
				return false
			}
			x++
			y++
		}
	}
	for i:=2;i<N;i++{
		var x,y int
		x = N - i
		for j:=0;j<i-1 && j<M-1;j++{
			if matrix[y][x] != matrix[y+1][x+1]{
				return false
			}
			x++
			y++
		}
	}
	return true
}

猜你喜欢

转载自blog.csdn.net/lolimostlovely/article/details/81245111
今日推荐