2319. 判断矩阵是否是一个 X 矩阵
题目:
如果一个正方形矩阵满足下述 全部 条件,则称之为一个 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 矩阵。
示例 2:
输入: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] <= 105
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/check-if-matrix-is-x-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
首先,我们先统计了矩阵的行列大小。
其次,我们进行遍历,如果一个元素行列相等或者行列相加等于矩阵的大小,那么就判断这个元素是否为0,如果是0,直接返回FALSE,如果不是,那么继续进行遍历。
最后,当一个元素行列不满足上述条件的时候,我们就判断这个元素是否不为0,如果不为0,那么就返回FALSE。当全部都遍历后没有返回FALSE,就返回TRUE即可。
代码:
class Solution {
public:
bool checkXMatrix(vector<vector<int>>& grid) {
int len = grid.size();
for(int i = 0;i < len;i++)
{
for(int j = 0;j < len;j++)
{
if(j == i || j == len-i-1)
{
if(grid[i][j] == 0)
return false;
}
else
if(grid[i][j])
return false;
}
}
return true;
}
};