美文网首页
LeetCode每日一题:矩阵找规律

LeetCode每日一题:矩阵找规律

作者: ShowMeCoding | 来源:发表于2023-01-30 22:54 被阅读0次

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

    如果一个正方形矩阵满足下述 全部 条件,则称之为一个 X 矩阵 :
    矩阵对角线上的所有元素都 不是 0
    矩阵中所有其他元素都是 0
    给你一个大小为 n x n 的二维整数数组 grid ,表示一个正方形矩阵。如果 grid 是一个 X 矩阵 ,返回 true ;否则,返回 false 。


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

    class Solution:
        def checkXMatrix(self, grid: List[List[int]]) -> bool:
            n = len(grid)
            # 对角线元素
            for i in range(n):
                for j in range(n):
                    if i == j or i + j == n-1:
                        if grid[i][j] == 0:
                            return False
                    else:
                        if grid[i][j] != 0:
                            return False
            return True
    
    • 解题思路
      对角线元素索引[row, col]的规律
      斜向下对角线元素索引:row = col
      斜向上对角线元素索引:row + col = n-1(n x n的矩阵)

    相关文章

      网友评论

          本文标题:LeetCode每日一题:矩阵找规律

          本文链接:https://www.haomeiwen.com/subject/iedvhdtx.html