字母异位词分组

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

示例:

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

说明:

  • 所有输入均为小写字母。
  • 不考虑答案输出的顺序。

解答:

public class Test {
	
	/**
	 * 字母异位词分组
	 * 思路:将每个单词按照字典升序排序(如:key变成eky)进行分组
	 * @param strs
	 * @return
	 */
	public List<List<String>> groupAnagrams(String[] strs) {
        Map<String, List<String>> map = new HashMap<String, List<String>>();
        for(String str: strs) {
        	//使用快速排序进行排序
        	String key = sortStr(str);
        	if(map.containsKey(key)) {
        		List<String> list = map.get(key);
        		list.add(str);
        	}else {
				List<String> list = new ArrayList<String>();
				list.add(str);
				map.put(key, list);
			}
        }
        
        List<List<String>> res = new ArrayList<List<String>>();
        for(String key: map.keySet()) {
        	res.add(map.get(key));
        }
		
		return res;
    }
	
	//快排进行排序
	public String sortStr(String str) {
		if(str == null || str == "") {
			return null;
		}
		char[] arr = str.toCharArray();
		quickSort(arr, 0, arr.length-1);
		return new String(arr);
	}
	
	//快排
	public void quickSort(char[] arr, int start, int end) {
		if(start >= end) {
			return ;
		}
		int left = start, right = end;
		//选取基准值(left)
		char temp = arr[left];
		while(left < right) {
			//从右往左,找到比基准值小的,填到基准值的坑1(arr[left])里
			while(left< right && arr[right] >= temp) {
				right--;
			}
			arr[left] = arr[right];
			//从左往右,找到比基准值大的,填到基准值的坑2(arr[right])里
			while(left<right && arr[left]<=temp) {
				left++;
			}
			arr[right] = arr[left];
		}
		//将基准值填到坑3里
		arr[left] = temp;
		//此时,一遍已完成,开始递归基准值左右子集
		quickSort(arr, start, left-1);
		quickSort(arr, left+1, end);
		
	}
	
	public static void main(String[] args) {
		Test test = new Test();
		String [] strs = new String[]{"eat", "tea", "tan", "ate", "nat", "bat"};
		System.out.println(test.groupAnagrams(strs));
	}
}

使用类库,精简代码

/**
 * 字母异位词分组
 * 思路:将每个单词按照字典升序排序(如:key变成eky)进行分组
 * @param strs
 * @return
 */
public List<List<String>> groupAnagrams(String[] strs) {
       Map<String, List<String>> map = new HashMap<String, List<String>>();
       for(String str: strs) {
       	char[] arr = str.toCharArray();
       	Arrays.sort(arr);
       	String key = String.valueOf(arr);
       	if(!map.containsKey(key)) {
       		map.put(key, new ArrayList<String>());
       	}
       	map.get(key).add(str);
       }
       
	return new ArrayList<List<String>>(map.values());
   }

你可能感兴趣的:(算法)