【C++刷题】力扣-#422-有效的单词方块

题目描述

给定一个字符串列表 words，每个单词由小写字母组成。我们要形成一个单词的正方形（即，第一行由 words[0] 组成，第二行由 words[1] 组成，依此类推，一次轮换），如果存在这样的排列，返回 true；否则，返回 false。
单词的正方形是指正方形中每行的单词必须由相同的字母组成，也就是说，如果 words[i] 在正方形中位于 words[j] 的正下方，那么 words[i] 和 words[j] 必须具有相同的长度，并且 words[i} 中的每个字母（数量上）都必须在 words[j] 中出现。

示例

示例 1:

输入: words = ["you","i","a","m"]
输出: true
解释: 
对于上述的 words，我们可以排列成如下的正方形：
you
i
a
m
可以验证单词是正方形的，因为 each row's letters can form the letters in the bottom row.

示例 2:

输入: words = ["abc","defgh","ijkl"]
输出: false
解释: 
这些单词不能形成正方形，因为 you would need 3 rows, which does not allow each letter to be repeated.

示例 3:

输入: words = ["ball","area","lead","lady"]
输出: false
解释: 
这些单词不能形成正方形，因为 there is no way to rearrange the letters to make a valid square.

题解

这个问题可以通过检查每个字母出现的次数来解决。

统计字母出现次数：创建一个哈希表来统计每个字母在所有单词中出现的次数。
检查每个单词的字母：对于列表中的每个单词，检查它的每个字母是否在哈希表中，并且哈希表中相应字母的计数是否足够。
检查正方形的可能性：确保所有单词的字母计数可以形成一个正方形，即所有行的字母数量可以匹配。

代码实现

bool validWordSquare(vector<string>& words) {
    if (words.empty()) return true;
    
    unordered_map<char, int> countMap;
    for (const auto& word : words) {
        for (char c : word) {
            countMap[c]++;
        }
    }
    
    for (const auto& word : words) {
        unordered_map<char, int> rowMap;
        for (char c : word) {
            rowMap[c]++;
        }
        
        for (char c : word) {
            if (countMap[c] < rowMap[c]) {
                return false;
            }
        }
    }
    
    return true;
}