力扣 2319. 判断矩阵是否是一个 X 矩阵

2319. 判断矩阵是否是一个 X 矩阵

如果一个正方形矩阵满足下述 全部 条件，则称之为一个 X 矩阵 ：

矩阵对角线上的所有元素都 不是 0
矩阵中所有其他元素都是 0

给你一个大小为 n x n 的二维整数数组 grid ，表示一个正方形矩阵。如果 grid 是一个 X 矩阵 ，返回 true ；否则，返回 false 。

涉及知识点：

二维矩阵的for循环

代码部分

class Solution:
    def checkXMatrix(self, grid: List[List[int]]) -> bool:
        for i in range(len(grid[0])):
            for j in range(len(grid)):
                if i==j or len(grid)-j-1==i :
                    if grid[i][j]==0:
                        return False
                else:
                    if (grid[i][j]!=0):
                        return False
        return True