对一个大小为 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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
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
216 ms 32.4 MB Python3