Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรมตรวจสอบว่าเมทริกซ์เป็น Toeplitz Matrix หรือไม่ใน Python

Toeplitz matrix: ทุกแนวทแยงมุม (ซ้ายบนไปขวาล่าง) มีค่าเท่ากันทุกตำแหน่ง

ตัวอย่าง: [[7,2,6], [3,7,2], [5,3,7]] → True (แนวทแยง: 7-7-7, 2-2, 6, 3-3, 5)

อัลกอริธึม

  • เช็คทุกเซลล์ (ยกเว้นแถวสุดท้ายและคอลัมน์สุดท้าย): matrix[i][j] == matrix[i+1][j+1]
  • ถ้าทุกคู่ตรงกัน → True

การนำไปใช้

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))  # True