작심삼일

[LeetCode] 766 | Toeplitz Matrix | Python 본문

스터디/코테

[LeetCode] 766 | Toeplitz Matrix | Python

yun_s 2022. 10. 31. 09:49
728x90
반응형

문제 링크: https://leetcode.com/problems/toeplitz-matrix/


문제

Given an m x n matrix, return true if the matrix is Toeplitz. Otherwise, return false.

A matrix is Toeplitz if every diagonal from top-left to bottom-right has the same elements.


조건

  • m == matrix.length
  • n == matrix[i].length
  • 1 <= m, n <= 20
  • 0 <= matrix[i][j] <= 99

내 풀이

Toepliz matrix가 되는 조건은 간단하다.

행렬의 모든 원소에 대해서 matrix[h][w] == matrix[h+1][w+1]이 성립되어야 한다.

이를 이용해서 해결하면 된다.


코드

class Solution:
    def isToeplitzMatrix(self, matrix: List[List[int]]) -> bool:
        H, W = len(matrix), len(matrix[0])
        
        for w in range(W-1):
            for h in range(H-1):
                if matrix[h][w] != matrix[h+1][w+1]:
                    return False
        
        return True
728x90
반응형
Comments