12. Group Anagrams

読了目安 約1分

HashMap / LeetCode 49 (Medium) — ソートした文字列をキーにしてアナグラムをまとめる。

問題

LeetCode 49. Group Anagrams（Medium / カテゴリ: HashMap）

文字列の配列 strs が与えられます。 アナグラムは、文字を並べ替えると互いに一致する単語の組です。 アナグラムどうしをグループにまとめ、グループの配列として返します。 グループの順序も、グループ内の単語の順序も自由です。

考え方

Swift 実装のポイント

• s.sorted() は [Character] を返します。キーにするには String(s.sorted()) で文字列に戻します。
• groups[key, default: []].append(s) で、キーが未登録のときの初期化を 1 行にまとめられます。
• 辞書の values の並び順は不定です。テストコードは各グループと全体をソートしてから比較します。

class Solution {
    func groupAnagrams(_ strs: [String]) -> [[String]] {
        var groups: [String: [String]] = [:]  // ソート済み文字列 → アナグラムの仲間
        for s in strs {
            let key = String(s.sorted())
            groups[key, default: []].append(s)
        }
        return Array(groups.values)
    }
}