题目

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

矩阵对角线上的所有元素都 不是 0
矩阵中所有其他元素都是 0
给你一个大小为 n x n 的二维整数数组 grid ，表示一个正方形矩阵。如果 grid 是一个 X 矩阵 ，返回 true ；否则，返回 false 。

示例 1：

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/check-if-matrix-is-x-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法

暴力破解

题目很简单，就是找规律的题目

class Solution:
    def checkXMatrix(self, grid: List[List[int]]) -> bool:
        count = len(grid)
        for row_index, row in enumerate(grid):
            for col_index, col in enumerate(row):
                if row_index == col_index or (count - row_index - 1) == col_index:
                    if col == 0:
                        return False
                else:
                    if col != 0:
                        return False
        return True