用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