字母异位词分组

package com.algorithm.code;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/*
给定一个字符串数组，将字母异位词组合在一起。字母异位词指字母相同，但排列不同的字符串。
 
示例:
 
输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]
所有输入均为小写字母。
不考虑答案输出的顺序。
 */
public class GroupAnagrams {
    public List<List<String>> groupAnagrams(String[] strs) {
        List<List<String>> result = new ArrayList<>();
        Map<String, List<String>> map = new HashMap<>();
        for (int i = 0; i <strs.length ; i++) {
            char[] chars = strs[i].toCharArray();
            Arrays.sort(chars);
            String newStrs = new String(chars);
            if (map.containsKey(newStrs)){
                List<String> list = map.get(newStrs);
                list.add(strs[i]);
                map.put(newStrs,list);
            }else {
                List<String> list = new ArrayList<>();
                list.add(strs[i]);
                map.put(newStrs,list);
            }
        }
        result.addAll(map.values());
        return result;
    }
 
    public static void main(String[] args) {
        GroupAnagrams ga = new GroupAnagrams();
        String[] strs = {"eat","tea","tan","ate","nat","bat"};
        System.out.println( ga.groupAnagrams(strs));
    }
}