用Python刷LeetCode必备知识点3 - collections.defaultdict()

在用Python刷LeetCode时,字典dict是最经常被用到的数据结构,但是对于访问dict,要是键值key不存在,就会有‘KeyError’的错,在写代码时就要先判断key是否存在,因此就要多写好几行代码来避免‘KeyError’。特别是类似使用列表字典,或者用于统计频率等。

比如要统计一个字符串中字符的频率,用普通的dict,代码如下:

freq = {}
for c in s:
    if c in freq:
        freq[c] += 1
    else:
        freq[c] = 1

为了避免‘KeyError’错,可以使用collections模块的defaultdict(),它的用法跟dict一样,但是使用defaultdict.defaultdict(),Python会为不存在的key赋一个初始值,从而避免‘KeyError’。比如以上用于统计字符频率的字典就可以初始化为freq = collections.defaultdict(int),对于不存在的字符初始值为0, 代码就可以简化为:

freq = collections.defaultdict(int)
for c in s:
    freq[c] += 1

实际应用:767. Reorganize String

上一篇:Apache RewriteCond %{REQUEST_FILENAME} 不起作用问题


下一篇:java中Collection集合、Iterator迭代器接口、for each循环遍历集合或对象