422. Valid Word Square

Given a sequence of words, check whether it forms a valid word square.

A sequence of words forms a valid word square if the kth row and column read the exact same string, where 0 ≤ k < max(numRows, numColumns).

Note:

  1. The number of words given is at least 1 and does not exceed 500.
  2. Word length will be at least 1 and does not exceed 500.
  3. Each word contains only lowercase English alphabet a-z.

Example 1:

Input:
[
  "abcd",
  "bnrt",
  "crmy",
  "dtye"
]

Output:
true

Explanation:
The first row and first column both read "abcd".
The second row and second column both read "bnrt".
The third row and third column both read "crmy".
The fourth row and fourth column both read "dtye".

Therefore, it is a valid word square.

Example 2:

Input:
[
  "abcd",
  "bnrt",
  "crm",
  "dt"
]

Output:
true

Explanation:
The first row and first column both read "abcd".
The second row and second column both read "bnrt".
The third row and third column both read "crm".
The fourth row and fourth column both read "dt".

Therefore, it is a valid word square.

Example 3:

Input:
[
  "ball",
  "area",
  "read",
  "lady"
]

Output:
false

Explanation:
The third row reads "read" while the third column reads "lead".

Therefore, it is NOT a valid word square.

Rust Solution

struct Solution;

impl Solution {
    fn valid_word_square(words: Vec<String>) -> bool {
        let n = words.len();
        for i in 0..n {
            let m = words[i].len();
            if m > n {
                return false;
            }
        }
        let mut a: Vec<Vec<u8>> = vec![vec![0; n]; n];
        for i in 0..n {
            for (j, b) in words[i].bytes().enumerate() {
                a[i][j] = b;
            }
        }
        for i in 0..n {
            for j in 0..n {
                if a[i][j] != a[j][i] {
                    return false;
                }
            }
        }
        true
    }
}

#[test]
fn test() {
    let words: Vec<String> = vec_string!["abcd", "bnrt", "crmy", "dtye"];
    assert_eq!(Solution::valid_word_square(words), true);
    let words: Vec<String> = vec_string!["abcd", "bnrt", "crm", "dt"];
    assert_eq!(Solution::valid_word_square(words), true);
    let words: Vec<String> = vec_string!["ball", "area", "read", "lady"];
    assert_eq!(Solution::valid_word_square(words), false);
    let words: Vec<String> = vec_string!["ball", "asee", "let", "lep"];
    assert_eq!(Solution::valid_word_square(words), false);
}

Having problems with this solution? Click here to submit an issue on github.