LeetCode 热题 100 之 49字母异位词分组

题目描述

给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
 
示例 1:

输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
示例 2:

输入: strs = [""]
输出: [[""]]
示例 3:

输入: strs = ["a"]
输出: [["a"]]

提示：
1 <= strs.length <= 104
0 <= strs[i].length <= 100
strs[i] 仅包含小写字母

思路

首先根据定义，可以知道，构成字母异位词的各个字母个数要一样才能分到一起。加上提示中说明只含有小写字母，因此可以用类似于哈希的方法，用a[10001][26]统计每个词的各字母个数。之后便可以用3层遍历的方法实现对给定的字母异位词进行划分。

代码

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> result;//返回结果
        int a[10001][26];//记录每个词的各个字母个数
        int b[10001];//记录每个词的字母个数
        for(int i = 0; i <10001; i++){
            b[i] = 0;
            for(int j = 0; j < 26; j++){
                a[i][j]=0;
            }
        }
        vector<string>::iterator it;//迭代器
        int p = 0;//记录一共有多少各词
        for(it=strs.begin();it!=strs.end();it++)
        {
            string str = *it;
            int len = str.length();//词的字母数
            b[p]=len;//记录第p+1个词的字母个数
            for(int i = 0; i < len; i++){
                a[p][str[i]-97]++;//记录字母数
            }
            p++;
        }

        bool divided[p];//记录词是否已经被划分
        for(int i = 0; i < p; i++){
            divided[i]=false;//初始化
        }
        for(int i = 0; i < p; i++){
            if(divided[i]==true){//如果这个词已经被划分完则跳过该词
                continue;
            }
            vector<string> stritem;//容器 存储划分到一起的字母异位词
            for(int j = 0;j < p;j++){
                //已划分或者对比的两个词的字母个数不同  则跳过
                if(divided[j]==true||b[i]!=b[j]){
                    continue;
                }
                int  k = 0;
                for(k = 0; k<26;k++){
                    if(a[i][k]!=a[j][k]){
                        break;// 对比两个词的各字母数,不同则跳出
                    }
                }
                if(k == 26){//两个词是字母异位词
                    stritem.push_back(strs[j]);//划分到一起.
                    divided[j]=true;//记录,已经分完
                }

            }
            result.push_back(stritem);//加入到最终结果中.
        }
        return result;

    }
};

python3

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        a = {}//字典
        for i in strs:
            key = "".join(sorted(i))//对字母进行排序得到字典的键
            if key not in a:
                a[key] = [i]//异位词列表则加入字典中,键值为排完序的字母.
            else:
                a[key].append(i)//加入到对应的异位词列表中
        return list(a.values())