LeetCode Anagrams

原题链接在这里：https://leetcode.com/problems/anagrams/

Given an array of strings, group anagrams together.

For example, given: ["eat", "tea", "tan", "ate", "nat", "bat"], 
Return:

[
  ["ate", "eat","tea"],
  ["nat","tan"],
  ["bat"]
]

Note:

1. For the return value, each inner list's elements must follow the lexicographic order.
2. All inputs will be in lower-case.

 

新的做法如下：与之前的字母排序不同，用getHash算出一个string 对应的hashcode. 用这个hash code当HashMap的key, 可以节省一部分时间。

 AC Java:

public class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        List<List<String>> res = new ArrayList<List<String>>();
        if(strs == null || strs.length == 0){
            return res;
        }
        Arrays.sort(strs); //error
        HashMap<Integer, List<String>> hm = new HashMap<Integer, List<String>>();
        for(int i = 0; i<strs.length; i++){
            int [] count = getCount(strs[i]);
            int hash = getHash(count);
            if(!hm.containsKey(hash)){
                List<String> item = new ArrayList<String>(); //error
                item.add(strs[i]);
                hm.put(hash, item);    
            }else{
                hm.get(hash).add(strs[i]);
            }
        }
        
        for(List<String> ls : hm.values()){
            res.add(ls);
        }
        return res;
    }
    
    private int[] getCount(String s){
        int [] count = new int[26];
        if(s == null || s.length() == 0){
            return count;
        }
        for(int i = 0; i<s.length(); i++){
            count[s.charAt(i) - 'a']++;
        }
        return count;
    }
    
    private int getHash(int [] count){
        int a = 378551;
        int b = 63689;
        int hash = 0;
        for(int i = 0; i<count.length; i++){
            hash = hash*a + count[i];
            a*=b;
        }
        return hash;
    }
}

  

做这道题让我想起刚才的 Valid Anagram 简直想的太麻烦了。

这道题的思路就是sort strs 数组中的每一个string，建立HashMap，key 是 sorted string, value 是 list of all original strings.

最后iterate hm，所有value 长度大于1的都是anagram。

Note：1. sort string 就用 string.toCharArray(), 然后用Arrays.sort(), 最后建立一个new String。这里注意是Arrays 而不是 Array。

    2. list 的长度用 list.size().

    3. 链接现有list 到一个新的list上可以直接用 mainList.addAll(subList) 来完成。

AC Java：

public class Solution {
    public List<String> anagrams(String[] strs) {
        if(strs == null || strs.length == 0)
            return null;
            
        List<String> res = new ArrayList<String>();
        
        HashMap<String,ArrayList<String>> hm = new HashMap<String,ArrayList<String>>();
        
        for(int i = 0; i < strs.length; i++){
            String key = sortStr(strs[i]);
            if(hm.containsKey(key)){
                hm.get(key).add(strs[i]);
            }else{
                ArrayList<String> valList = new ArrayList<String>();
                valList.add(strs[i]);
                hm.put(key, valList);
            }
        }
        
        for(String s : hm.keySet()){
            if(hm.get(s).size() > 1){
                res.addAll(hm.get(s));
            }
        }
        return res;        
        
    }
    
    private String sortStr(String s){
        char[] temp = s.toCharArray();
        Arrays.sort(temp); //error
        String str = new String(temp);
        return str;
    }
}