【原】LeetCode刷题实战422：有效的单词方块

程序IT圈 2021-10-28

展开全文

算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选！

今天和大家聊的问题叫做 有效的单词方块，我们先来看题面：

https:///problems/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:

The number of words given is at least 1 and does not exceed 500.

Word length will be at least 1 and does not exceed 500.

Each word contains only lowercase English alphabet a-z.

给你一个单词序列，判断其是否形成了一个有效的单词方块。

有效的单词方块是指此由单词序列组成的文字方块的第 k 行和第 k 列 (0 ≤ k < max(行数, 列数)) 所显示的字符串完全相同。

注意：

给定的单词数大于等于 1 且不超过 500。

单词长度大于等于 1 且不超过 500。

每个单词只包含小写英文字母 a-z。

示例

示例 1：
输入：
[
  "abcd",
  "bnrt",
  "crmy",
  "dtye"
]
输出：
true
解释：
第 1 行和第 1 列都是 "abcd"。
第 2 行和第 2 列都是 "bnrt"。
第 3 行和第 3 列都是 "crmy"。
第 4 行和第 4 列都是 "dtye"。
因此，这是一个有效的单词方块。

示例 2：
输入：
[
  "abcd",
  "bnrt",
  "crm",
  "dt"
]
输出：
true
解释：
第 1 行和第 1 列都是 "abcd"。
第 2 行和第 2 列都是 "bnrt"。
第 3 行和第 3 列都是 "crm"。
第 4 行和第 4 列都是 "dt"。
因此，这是一个有效的单词方块。

示例 3：
输入：
[
  "ball",
  "area",
  "read",
  "lady"
]
输出：
false
解释：
第 3 行是 "read" ，然而第 3 列是 "lead"。
因此，这不是一个有效的单词方块。

解题

检查单词长度是否等于单词个数，有长的，直接返回false，如果比较短的，补上空格。

class Solution {
public:
    bool validWordSquare(vector<string>& words) {
      int m = words.size(), i, j;
      for(i = 0; i < m; ++i)
      {
        if(words[i].size() > m)
          return false;
        if(words[i].size() < m)
          words[i] += string(m-words[i].size(),' ');
      }
    for(i = 0; i < m; ++i)
    {
      for(j = 0; j < m; ++j)
      {
        if(words[i][j] != words[j][i])
          return false;
      }
    }
    return true;
    }
};