LeetCode OJ算法题（四十八）：Anagrams

最新推荐文章于 2022-03-05 19:28:52 发布

CaptainStanley

最新推荐文章于 2022-03-05 19:28:52 发布

阅读量406

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/op_yu/article/details/38406045

版权

LeetCode 专栏收录该内容

80 篇文章 0 订阅

订阅专栏

题目：

Given an array of strings, return all groups of strings that are anagrams.

Note: All inputs will be in lower-case.

解法：

所谓的anagrams是指通过颠倒字符串顺序能够生成的字符串对，比如abc与acb与bac等

因此，一开始我使用HashMap存储每个string的字符，再遍历他后面的字符串，对每个字符串中的字符在map中查找，如果刚好匹配，就把这个字符串和一开始作为字典的字符串加入返回结果中，代码如下：

public static List<String> anagrams(String[] strs) {
        List<String> ret = new ArrayList<String>();
        for(int i=0;i<strs.length-1;i++){
        	boolean shouldAdd = false;
        	if(strs[i] == null) continue;
        	HashMap<Character, Integer> alphaMap = new HashMap<Character, Integer>();
        	for(int k=0;k<strs[i].length();k++){
        		if(alphaMap.get(strs[i].charAt(k)) == null){
        			alphaMap.put(strs[i].charAt(k), 1);
        		}
        		else{
        			alphaMap.put(strs[i].charAt(k), alphaMap.get(strs[i].charAt(k))+1);
        		}
        	}
        	for(int j=i+1;j<strs.length;j++){
        		if(strs[j] == null || strs[i].length() != strs[j].length()) continue;
        		HashMap<Character, Integer> cloneMap = (HashMap<Character, Integer>) alphaMap.clone();
        		int k=0;
        		for(;k<strs[j].length();k++){
        			if(cloneMap.get(strs[j].charAt(k))==null || cloneMap.get(strs[j].charAt(k))==0) break;
        			cloneMap.put(strs[j].charAt(k), cloneMap.get(strs[j].charAt(k))-1);
        		}
        		if(k == strs[j].length()){
        			shouldAdd = true;
        			ret.add(strs[j]);
        			strs[j] = null;
        		}
        	}
        	if(shouldAdd) ret.add(strs[i]);
        }
        return ret;
    }

结果TLE了，因为遍历太多了，时间复杂度为O（n^2*m）

看到网上其他的思路，发现可以通过对每个字符串对应的char数组进行排序，这样只要是anagrams，他们排序后的字符串一定是一样的，因此我们可以使用这个字符串最为HashMap的Key，而之所以不用set是因为我们还要把作为字典的字符串视情况加入返回结果中，因此map的值用<String , Boolean>的键值对。

代码如下：

public static List<String> anagramsII(String[] strs){
		class Pair{
			String str;
			boolean shouldAdd;
			public Pair(String s, boolean b){
				str = s;
				shouldAdd = b;
			}
		}
		List<String> ret = new ArrayList<String>();
		HashMap<String,Pair> map = new HashMap<String,Pair>();
		for(int i=0;i<strs.length;i++){
			char[] tmp = strs[i].toCharArray();
			Arrays.sort(tmp);
			String tmps = String.valueOf(tmp);
			if(map.containsKey(tmps)){
				ret.add(strs[i]);
				map.put(tmps, new Pair(map.get(tmps).str,true));
			}
			else{
				map.put(tmps,new Pair(strs[i], false));
			}
		}
		Iterator it = map.entrySet().iterator();
		while(it.hasNext()){
			java.util.Map.Entry<String, Pair> entry = (java.util.Map.Entry<String, Pair>)it.next();
			if(entry.getValue().shouldAdd){
				ret.add(entry.getValue().str);
			}
		}
		return ret;
	}