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

输入：grid = [[2,0,0,1],[0,3,1,0],[0,5,2,0],[4,0,0,2]]
输出：true
解释：矩阵如上图所示。
X 矩阵应该满足：绿色元素（对角线上）都不是 0 ，红色元素都是 0 。
因此，grid 是一个 X 矩阵。

输入：grid = [[5,7,0],[0,3,1],[0,5,0]]
输出：false
解释：矩阵如上图所示。
X 矩阵应该满足：绿色元素（对角线上）都不是 0 ，红色元素都是 0 。
因此，grid 不是一个 X 矩阵。
 
提示：
n == grid.length == grid[i].length
3 <= n <= 100
0 <= grid[i][j] <= 10^5

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