49. 字母异位词分组(Medium)
判断两个词是不是字母异位词最简单的方法就是排序,字符串是可以排序的,排序返回的是一个list。当然也可以使用哈希表判断每一个词出现的个数,这也时间复杂度会低,但是代码显得麻烦。
本题是要把所有的字母异位词放到一起,最暴力的方法就是两层循环,每次比对两个字符串,对他们进行排序,看看一样不一样,一样的话说明属于同一个字母异位词。这样做最好不要使用for循环,因为for循环需要从头遍历到尾,内存循环可能会提前将后面的一个词归于前面某个词的list里,那么外层循环在遇到这个词的时候是不需要执行操作的,所以还需要维护一个visited数组来防止出现重复的情况。
而最好的方法就是每次找到一个字母异位词,就把他从strs里面删除,这也最后strs会成为空的,这种用while循环更合适。
而另外一种更高效的方法就是使用字典,字典的键就是排序后的字符串,因为字母异位词排序后结果一样,所以只需要按照排序后的字符串作为建,把每一个字符串放到对应的value里面,最后再把value合并一下就行了。代码如下:
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
dic = collections.defaultdict(list)
# 把排序后相同的s放到一个list里
for s in strs:
dic[''.join(sorted(s))].append(s)
res = [v for v in dic.values()]
return res
还有一种暴力一点的方法,就是提前对每个字符串进行排序,之后两层循环遍历的时候就不需要再次排序了。
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
strs_sort = [sorted(s) for s in strs]
n = len(strs)
visited = set()
res = []
for i in range(n):
if i not in visited:
tmp = [strs[i]]
visited.add(i)
for j in range(i, n):
if j not in visited and strs_sort[i] == strs_sort[j]:
visited.add(j)
tmp.append(strs[j])
res.append(tmp)
return res
但是这种做法很耗时。
954. 二倍数对数组
思路很简单,就是看一个数,是否存在他的二倍或者一半的数,然后把所有数字凑成这样的pair。但是问题是:
- 如果从前往后遍历,遇到一个数,是优先找他的二倍还是一半呢?(比如[1, 2, 0, 0, 4, 8])。如果把2和4凑到一块去了,那最后就没办法凑出来三对,结果就会返回False。但是实际上是可以凑的,只要把4和2凑一起。
如果排序呢?加入全是正数,排完序之后,就可以优先找二倍的数了。但是里面有正有负,我们可以使用绝对值进行排序。
class Solution:
def canReorderDoubled(self, arr: List[int]) -> bool:
# 只要看这n个数能否分成n//2份,每一份都是两个数,其中一个数是另一个的二倍
n = len(arr)
dic = collections.defaultdict(int)
for i in range(n):
dic[arr[i]] += 1 # 统计每个数出现的次数
# 如果数组中存在0,必须是偶数个0才可以
if dic[0] % 2 != 0:
return False
# 对arr按照绝对值排序
for x in sorted(dic, key=abs):
# 优先处理绝对值较小的数
if dic[2*x] < dic[x]:
return False
dic[2*x] -= dic[x]
return True
还有一些其他简单的题,这里就不放了。