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

LeetCode:2319.判断矩阵是否是一个X矩阵
题目如下：

如题可分析得出，该矩阵对角线坐标可以表示为：‘\’方向，i=j；‘/’方向，i+j+1=矩阵边长（len(grid)）。在以上两种情况中，不能为0。同时，在不符合上述两种情况中必须为0。
具体代码如下：

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