1. 程式人生 > >Leetcode:Toeplitz Matrix

Leetcode:Toeplitz Matrix

and integer object 相同 element leetcode str only column

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.

Input: matrix = [[1,2,3,4],[5,1,2,3],[9,5,1,2]]
Output: True
Explanation:
1234
5123
9512

In the above grid, the diagonals are "[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]", and in each diagonal all elements are the same, so the answer is True.

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].

思路
這純粹是一道找規律的題目,每一行的前n-1個數和下一行的後n-1個數相同

class Solution(object):
    def isToeplitzMatrix(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: bool
        """
        flag = False
        for key,value in enumerate(matrix):
            if key + 1< len(matrix):
                if value[:-1] == matrix[key+1][1:]:
                    continue
                else:
                    flag = False
                    break
            else:
                flag = True
        return flag

Leetcode:Toeplitz Matrix