Python教程

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

本文主要是介绍用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

这篇关于用Python刷LeetCode必备知识点3 - collections.defaultdict()的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航