LeetCode 2133. 检查是否每一行每一列都包含全部整数

Michael阿明

发布于 2022-03-10 09:57:28

57300

代码可运行

文章被收录于专栏：Michael阿明学习之路Michael阿明学习之路

运行总次数：0

代码可运行

文章目录
1. 题目
2. 解题

1. 题目

对一个大小为 n x n 的矩阵而言，如果其每一行和每一列都包含从 1 到 n 的全部整数（含 1 和 n），则认为该矩阵是一个有效矩阵。

给你一个大小为 n x n 的整数矩阵 matrix ，请你判断矩阵是否为一个有效矩阵：如果是，返回 true ；否则，返回 false 。

示例 1：

输入：matrix = [[1,2,3],[3,1,2],[2,3,1]]
输出：true
解释：在此例中，n = 3 ，每一行和每一列都包含数字 1、2、3 。
因此，返回 true 。

示例 2：

输入：matrix = [[1,1,1],[1,2,3],[1,2,3]]
输出：false
解释：在此例中，n = 3 ，但第一行和第一列不包含数字 2 和 3 。
因此，返回 false 。
 
提示：
n == matrix.length == matrix[i].length
1 <= n <= 100
1 <= matrix[i][j] <= n

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

2. 解题

按题意对行列求和，并判断不同的数的个数是不是为 n 个

class Solution:
    def checkValid(self, matrix: List[List[int]]) -> bool:
        n = len(matrix)
        import numpy as np
        matrix = np.array(matrix)
        s = (n+1)*n//2
        for i in range(n):
            if sum(matrix[i, :]) != s or len(set(matrix[i, :])) != n:
                return False
        for i in range(n):
            if sum(matrix[:, i]) != s or len(set(matrix[:, i])) != n:
                return False
        return True