leetcode之哈希表五: 检查是否每一行每一列都包含全部整数-CSDN博客

本文链接：https://blog.csdn.net/qq_32523587/article/details/122444442

原题:

力扣https://leetcode-cn.com/problems/check-if-every-row-and-column-contains-all-numbers/

一、题目要求

对一个大小为 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

二、解题

package com.leetcode.hash;

import java.util.HashSet;

public class Solution17 {
    public static void main(String[] args) {
        int[][] matrix = {{1,2,3},{3,1,2},{2,3,1}};

        System.out.print(checkValid(matrix));
    }


    public static boolean checkValid(int[][] matrix) {
        int len = matrix.length;
        HashSet<Integer> set = new HashSet<Integer>();

        // 遍历每一行
        for (int i = 0; i < len; i++) {
            initSet(set, len);
            for (int j = 0; j < len; j++) {
                boolean b = set.remove(matrix[i][j]);
                if (!b) {
                    return false;
                }
            }
        }

        // 遍历每一列
        for (int i = 0; i < len; i++) {
            initSet(set, len);
            for (int j = 0; j < len; j++) {
                boolean b = set.remove(matrix[j][i]);
                if (!b) {
                    return false;
                }
            }
        }

        return true;
    }

    public static void initSet(HashSet<Integer> set, int len) {
        for (int i = 1; i <= len; i++) {
            set.add(i);
        }
    }
}