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

2022/2/7 14:13:30

本文主要是介绍用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()的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程