2133. Check if Every Row and Column Contains All Numbers

An n x n matrix is valid if every row and every column contains all the integers from 1 to n (inclusive).

Given an n x n integer matrix matrix, return true if the matrix is valid. Otherwise, return false.

 

Example 1:

Input: matrix = [[1,2,3],[3,1,2],[2,3,1]]
Output: true
Explanation: In this case, n = 3, and every row and column contains the numbers 1, 2, and 3.
Hence, we return true.

Example 2:

Input: matrix = [[1,1,1],[1,2,3],[1,2,3]]
Output: false
Explanation: In this case, n = 3, but the first row and the first column do not contain the numbers 2 or 3.
Hence, we return false.

 

Constraints:

  • n == matrix.length == matrix[i].length
  • 1 <= n <= 100
  • 1 <= matrix[i][j] <= n

2133. Check if Every Row and Column Contains All Numbers
struct Solution;

impl Solution {
    fn check_valid(matrix: Vec<Vec<i32>>) -> bool {
        let n = matrix.len();
        for i in 0..n {
            let mut col = vec![false; n];
            for j in 0..n {
                let x = matrix[i][j];
                if (1..=n as i32).contains(&x) {
                    let k = (x - 1) as usize;
                    col[k] = true;
                } else {
                    return false;
                }
            }
            for j in 0..n {
                if !col[j] {
                    return false;
                }
            }
        }
        for i in 0..n {
            let mut row = vec![false; n];
            for j in 0..n {
                let x = matrix[j][i];
                if (1..=n as i32).contains(&x) {
                    let k = (x - 1) as usize;
                    row[k] = true;
                } else {
                    return false;
                }
            }
            for j in 0..n {
                if !row[j] {
                    return false;
                }
            }
        }

        true
    }
}

#[test]
fn test() {
    let matrix = vec_vec_i32![[1, 2, 3], [3, 1, 2], [2, 3, 1]];
    let res = true;
    assert_eq!(Solution::check_valid(matrix), res);
    let matrix = vec_vec_i32![[1, 1, 1], [1, 2, 3], [1, 2, 3]];
    let res = false;
    assert_eq!(Solution::check_valid(matrix), res);
}