变位词组

面试题 10.02. 变位词组

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

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

示例:

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

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

 

python解法:

class Solution:

    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:

        tmp=[]

        for i in range(len(strs)):

            tmp.append(''.join(sorted(strs[i])))

        res=[]

        dic={}

        for i in range(len(tmp)):

            if tmp[i] not in dic:

                res.append([strs[i]])

                dic[tmp[i]]=len(res)-1

            else:

                res[dic[tmp[i]]].append(strs[i])

        return res

首先排序后再对应存储,原先踩过的坑,在res放置时要注意放一维列表进去,其次dic哈希表存储的是对应组别存放的下标,如果已经出现过的话找到对应位置插进去,同时用.join()格式连接字符串。

 

你可能感兴趣的:(变位词组)