字母异位词分组

2019-09-27  本文已影响0人  Haward_

题目:
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]

思想:
定义一个int[] arr = new int[26];的字母表,加‘#’转成string做key的映射,即所有的相同分组都有一样的key。

import java.util.HashMap;
import java.util.ArrayList;
class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        List<List<String>> res = new ArrayList<List<String> >();
        if(strs.length==0) {
            return res;
        }
        
        HashMap<String,List> map = new HashMap<>();
        for(int i=0;i<strs.length;i++) {
            String str = strs[i];
            int[] arr = new int[26];
            for(Character c : str.toCharArray()) {
                arr[c-'a'] += 1;
            }
            StringBuffer sb = new StringBuffer();
            for(int n : arr) {
                sb.append(n);
                sb.append('#');
            }
            String key = sb.toString();
            if(map.get(key)==null) {
                map.put(key,new ArrayList<String>());
                map.get(key).add(str);
            }else {
                map.get(key).add(str);
            }
        }
        for(String key : map.keySet()) {
            res.add(map.get(key));
        }
        return res;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读