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

题目

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

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

思路

根据数据范围，可以直接暴力

代码

class Solution {
public:
    bool checkXMatrix(vector<vector<int>>& grid) {
        int n = grid.size();
        for(int i = 0; i <n; i++){
            for(int j = 0; j < n; j++){
                if(i == j || i + j == n - 1){
                    if(!grid[i][j])
                        return false;
                }else if (grid[i][j])
                    return false;
            }
        }
        return true;
    }
};