1. 程式人生 > >leetcode:49. 字母異位詞分組(java)

leetcode:49. 字母異位詞分組(java)

package LeetCode;

import java.util.*;

/*
給定一個字串陣列,將字母異位詞組合在一起。字母異位詞指字母相同,但排列不同的字串。

示例:

輸入: ["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>> res = new ArrayList<>();
        Map<String, List<String>> m = new HashMap<>();
        for (int i = 0; i <strs.length ; i++) {
            char[] r= strs[i].toCharArray();
            Arrays.sort(r);
            String k=new String(r);
            if (m.containsKey(k)){
                List<String> p=m.get(k);
                p.add(strs[i]);
                m.put(k,p);
            }else {
                List<String> p=new ArrayList<>();
                p.add(strs[i]);
                m.put(k,p);
            }
        }
        res.addAll(m.values());
        return res;
    }

    public static void main(String[] args) {
        GroupAnagrams s=new GroupAnagrams();
        String[] a={"eat","tea","tan","ate","nat","bat"};
        System.out.println( s.groupAnagrams(a));
    }
}