【数据结构与算法】力扣 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] 仅包含小写字母

分析解答

lodash 里面自带一个方法，可以根据规则对字符串数组进行分组，而分组后的 value 刚好是我们需要的结果。

import _ from 'lodash'
/**
 * @param {string[]} strs
 * @return {string[][]}
 */
var groupAnagrams = function(strs) {
    // 先对每个字符串进行排序 作为分组规则（函数返回值为 key） 值为满足规则的数组元素组成的数组 然后进行分组
    /*
     * 例如：_.groupBy([6.1, 4.2, 6.3], Math.floor); // => { '4': [4.2], '6': [6.1, 6.3] }
     */
    const g = _.groupBy(strs, str => str.split('').sort().join(''))
    console.log(g)
    // 返回 value 组成的数组
    return Object.values(g)
};
// 测试代码
console.log(groupAnagrams(["eat", "tea", "tan", "ate", "nat", "bat"]))

思路拓展

当然不用 lodash 也可以，我们自己实现一个分组，相当于把以上 lodash 的原理实现了一遍。

/**
 * @param {string[]} strs
 * @return {string[][]}
 */
var groupAnagrams = function(strs) {
    let res = {}
    for (let str of strs) {
        let arr = str.split('').sort().join()
        if (res[arr]) {
            res[arr].push(str)
        } else {
            res[arr] = [str]
        }
    }
    return Object.values(res)
};
// 测试代码
console.log(groupAnagrams(["eat", "tea", "tan", "ate", "nat", "bat"]))