题目要求:
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
input:[“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
output:[[‘eat’, ‘tea’, ‘ate’], [‘tan’, ‘nat’], [‘bat’]]
思路1
前两天写回溯法写魔怔了。。看到题第一反应是用回溯法。结果超时了
事实证明算法需要针对题目具体问题具体分析,不能全凭一套模板
但还是想记录一下整个回溯的思考过程,也算是第一个独立完整回溯出来的较为复杂的例子。
回溯条件:当前的target已经对比过整个字符串
终止条件:列表里的所有单词都已经做个比较
class Solution:
def same_letter(self, get, target):
lens = len(get)
lens_1 = len(target)
if lens != lens_1:
return False
for i in range(lens):
if get[i] not in target:
return False
else:
target = target.replace(get[i], "", 1)
return True
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
def helper(board, temp):
if len(board) == 0:
return True
target = board[0]
i = 0
while (i < len(board)): # 历遍整个board 寻找和target相同的单词
if self.same_letter(board[i], target):
temp += [board[i]]
board.remove(board[i])
else:
i += 1
res.append(temp)
helper(board, [])
res = []
helper(strs, [])
return res
思路2
需要确定一种方法,能够将子集用另一种方法唯一表示。这样另一种表示方法结果一致的,则是异位词。
这里参考题解用质数表示26个字母,这样不同字母的组合乘积必然不同,乘积
相同的组合必然是由相同字母组成的
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
from collections import defaultdict
prime = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103]
lookup = defaultdict(list)
for _str in strs:
key_val = 1
for s in _str:
key_val *= prime[ord(s) - 97]
lookup[key_val].append(_str)
return list(lookup.values())