> 文章列表 > 《程序员面试金典(第6版)》面试题 10.02. 变位词组

《程序员面试金典(第6版)》面试题 10.02. 变位词组

《程序员面试金典(第6版)》面试题 10.02. 变位词组

题目描述

编写一种方法,对字符串数组进行排序,将所有变位词组合在一起。变位词是指字母相同,但排列不同的字符串。

注意:本题相对原题稍作修改

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[["ate","eat","tea"],["nat","tan"],["bat"]
]

说明:

  • 所有输入均为小写字母。
  • 不考虑答案输出的顺序。

解题思路与代码

这道题是一道中等难度的题。考察的知识点我觉得主要是哈希映射。对于C++语言来讲,需要你对map或者是unordered_map容器的一些操作有基本的了解。这道题就不怎么难了。

方法一,使用哈希映射 + unordered_map容器解决问题

  • 在这道题中,我们需要维护一个unordered_map<string,vector<string>> map,其中,单词的字典排序作为map的key,而符合这个key的单词的一系列重组单词的集合,我们把它作为map的value。

  • 理解了上面一句话后,这道题就很简单了。

  • 我们先遍历题目给的这个vector<string>& strs,将它的每一个单词拿出来,先进行字典排序,然后去查找这个字典排序后的单词是否已经作为key在map中出现了,如果没有出现,则添加这个key,并且把这个单词未被排序之前的状态,作为一个value传递进去,比如这样:map[word] = {strs[i]};。如果出现了,我们就把这个单词添加到key所对应的value中去。比如这样:map[word].push_back(strs[i]);

  • 最后,我们再遍历一波map,把map中的每一个key所对应的value,作为一个vector,添加到最后的结果集中

具体的代码示例长这样:

class Solution {
public:vector<vector<string>> groupAnagrams(vector<string>& strs) {unordered_map<string,vector<string>> map;for(int i = 0; i < strs.size(); ++i){string word = strs[i];sort(word.begin(),word.end());if(map.find(word) == map.end()){map[word] = {strs[i]};}else{map[word].push_back(strs[i]);}}vector<vector<string>> result;for(auto& a : map){result.push_back(a.second);}return result;}
};

在这里插入图片描述

复杂度分析

时间复杂度:

  • 遍历 strs 中的每个字符串,时间复杂度为 O(N),其中 N 为 strs 中的字符串数量。
    对每个字符串进行排序,时间复杂度为 O(K * log(K)),其中 K 为字符串的最大长度。
    使用哈希表进行查找和插入操作,时间复杂度为 O(1)(平均情况下)。
    因此,总的时间复杂度为 O(N * K * log(K))。

空间复杂度:

  • unordered_map 存储了所有输入字符串,空间复杂度为 O(N)。
    输出结果 vector<vector> result,空间复杂度为 O(N)。
    因此,总的空间复杂度为 O(N)。

总结

  • 这道题目在考察使用哈希映射的能力以及对 C++ 中 map 和 unordered_map 等容器的操作是否熟悉。

  • 我们可以通过设计合适的哈希函数和键值对存储,可以有效地解决这类问题。