LeetCode Toeplitz Matrix

766. Toeplitz Matrix
A matrix is Toeplitz if every diagonal from top-left to bottom-right has the same element.
Now given an M x N matrix, return True if and only if the matrix is Toeplitz.
Example 1:
Input:
matrix = [
[1,2,3,4],
[5,1,2,3],
[9,5,1,2]
]
Output: True
Explanation:
In the above grid, the diagonals are:
“[9]”, “[5, 5]”, “[1, 1, 1]”, “[2, 2, 2]”, “[3, 3]”, “[4]”.
In each diagonal all elements are the same, so the answer is True.

Example 2:
Input:
matrix = [
[1,2],
[2,2]
]
Output: False
Explanation:
The diagonal “[1, 2]” has different elements.

Note:
matrix will be a 2D array of integers.
matrix will have a number of rows and columns in range [1, 20].
matrix[i][j] will be integers in range [0, 99].
Follow up:
What if the matrix is stored on disk, and the memory is limited such that you can only load at most one row of the matrix into the memory at once?
What if the matrix is so large that you can only load up a partial row into the memory at once?
解題關鍵:
判斷本行的[i][j]位置的元素與下一行的[i+1][j+1]元素相同,並且只判斷到[M-1]×[N-1]行,最後一個元素只有一個不用判斷。

class Solution(object):
    def isToeplitzMatrix(self, matrix):
        m = len(matrix)
        n = len(matrix[0])
        for i in range(m-1):
            for j in range(n-1):
                if matrix[i][j] == matrix[i+1][j+1]:
                    continue
                else:
                    return False
        return True             
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章