LeetCode - 字謎分組【Java | LeetCode中級算法】

 

給定一個字符串數組，將字母異位詞組合在一起。字母異位詞指字母相同，但排列不同的字符串。

示例:

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

說明：

• 所有輸入均爲小寫字母。
• 不考慮答案輸出的順序。

可以利用Map，將排序後的字符串作爲key

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
		HashMap<String, List<String>> hashMap = new HashMap<>();

        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String key = new String(chars);

            if (!hashMap.containsKey(key)) {
                hashMap.put(key, new ArrayList<>());
            }
            hashMap.get(key).add(str);
        }

        return new ArrayList<>(hashMap.values());
	}
}ew ArrayList<>(hashMap.values());

 開始的寫法，太慢太慢了

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
		List<List<String>> list = new ArrayList<>();
		// 一個詞一個詞遍歷
		for (String str : strs) {
			boolean flag = false;
			for (List<String> li : list) {
				if (isSame(li.get(0), str)) {
					li.add(str);
					flag = true;
					break;
				}
			}
			if (!flag) {
				List<String> newArr = new ArrayList<>();
				newArr.add(str);
				list.add(newArr);
			}
		}
		return list;

	}

	/**
	 * 判斷兩個字符串是否爲異位字符串
	 * @param str
	 * @return
	 */
	public boolean isSame(String str1, String str2) {
		// 長度不同，返回false
		if (str1.length() != str2.length()) {
			return false;
		}

		int[] arr1 = new int[26];
		
		char[] chArr1 = str1.toCharArray();
		for (char i : chArr1) {
			arr1[i - 'a']++;
		}
		
		int[] arr2 = new int[26];
		char[] chArr2 = str2.toCharArray();
		for (char i : chArr2) {
			arr2[i - 'a']++;
		}
		// 遍歷判斷
		for (int i = 0; i < arr1.length; i++) {
			// 存在不同字母，返回false
			if (arr1[i] != arr2[i]) {
				return false;
			}
		}
		return true;
	}
}