반응형
Notice
Recent Posts
Recent Comments
- Today
- Total
작심삼일
[LeetCode] 766 | Toeplitz Matrix | Python 본문
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