Chương trình kiểm tra ma trận Toeplitz trong Python

Cho ma trận M, kiểm tra xem có phải ma trận Toeplitz không. Ma trận Toeplitz khi mọi đường chéo từ trái sang phải có cùng giá trị.

Ví dụ, ma trận:

Các bước giải quyết:

• Duyệt từng hàng i trừ hàng cuối, từng cột j trừ cột cuối:
  • Nếu matrix[i][j] != matrix[i+1][j+1], trả về False.
• Trả về True.

Ví dụ minh họa

class Solution:
    def solve(self, matrix):
        for i in range(len(matrix) - 1):
            for j in range(len(matrix[0]) - 1):
                if matrix[i][j] != matrix[i+1][j+1]:
                    return False
        return True

ob = Solution()
matrix = [
    [7, 2, 6],
    [3, 7, 2],
    [5, 3, 7]
]
print(ob.solve(matrix))

Đầu vào

[[7, 2, 6], [3, 7, 2], [5, 3, 7]]

Đầu ra

True