Python刷题常用数据结构与函数总结

在刷题中,经常会用到一些常用的数据结构,自己实现费事费力,而且由于语言的原因,运行速度也比较慢。还好官方提供了很多方便的实现。

list

list,用[]表示。类似于C++的vector,Java的ArrayList,是一个可变长度的连续数组。便于插入删除最后一个元素。

常用操作:

  • append,在之后增加一个元素
  • pop,取出最后一个元素。pop(0)可以取出第一个元素,但是一般不推荐用这个来实现队列,因为取出第一个元素后需要复制整个数组。
  • __len__,获取长度。一般通过len(a)的形式调用。

tuple

元组,用()表示。表示多个元素的组合。如(1, 2, 3)

list就是一个栈,操作用append,pop,top指令可以用x[-1]替代

队列

list也可以当队列。操作用append,pop(0),front指令可以用x[0]替代。一般不推荐。建议使用deque当队列。后续介绍。

dict

dict是字典,用{}表示,如{k: v}。类似于HashMap。

常用操作:

  • keys,获取key的列表。
  • in,判断key是否在dict。如if a in d:
  • items,迭代器。如for k, v in d.items():
  • __len__,获取长度。一般通过len(a)的形式调用。
  • get(x, y),获取key为x的元素,如果x不存在,返回y。
  • setdefault(x, y),获取key为x的元素,如果x不存在,设置d[x] = y。常用于d.setdefault(k, []).append(1)

set

无序集合。用{}表示,如{a, b, c}。类似于C++的 unordered_set

常用操作:

  • add 增加一个元素
  • remove 删除一个元素
  • in判断元素是否在set中。如if a in s:

defaultdict

带有默认值的dict。类似于C++的unordered_map。获取不存在的元素时,会自动创建。用法:

from collections import defaultdict

d = defaultdict(list)
d['x'].append(1)

sums = defaultdict(int)
sums['apple'] += 1

deque

双头队列。操作前需要from collections import deque

常用操作:

  • append,在之后(右边)增加一个元素
  • popleft,pop最左边的元素
  • appendleft,在开头(左边)增加一个元素
  • pop,pop最右边的元素

heapq

小根堆。语法比较诡异,用于扩展list来支持堆操作。一般结合tuple来实现复杂的排序逻辑。

常用操作:

  • heapq.heapify(list),把一个list转化为一个heap,这是一个原地操作,即只影响list中元素的顺序。
  • heapq.heappush(heap, item) push一个元素
  • heapq.heappop(heap) pop一个元素
  • heapq.nlargest(n, iterable),返回迭代器中最大的n个元素
  • heapq.nsmallest(n, iterable), 返回迭代器中最小的n个元素

堆排序方法:

>>> from heapq import *
>>> def heapsort(iterable):
...     h = []
...     for value in iterable:
...         heappush(h, value)
...     return [heappop(h) for i in range(len(h))]
...
>>> heapsort([1, 3, 5, 7, 9, 2, 4, 6, 8, 0])
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

使用dijkstra算法求最小路径和:

# 给定一个包含非负整数的 m x n 网格,请找出一条从左上角到右下角的路径,使得路径上的数字总和为最小。
import heapq

class Solution:
    def minPathSum(self, grid: List[List[int]]) -> int:
        q = []
        q.append((0, 0, 0))

        m, n = len(grid), len(grid[0])
        dists = [[-1 for i in range(n)] for j in range(m)]

        while len(q) > 0:
            d, i, j = heapq.heappop(q)

            if dists[i][j] != -1:
                continue

            dists[i][j] = d

            if i < m - 1:
                heapq.heappush(q, (d + grid[i][j], i + 1, j))

            if j < n - 1:
                heapq.heappush(q, (d + grid[i][j], i, j + 1))

        return dists[m - 1][n - 1] + grid[m - 1][n - 1]

lru_cache

lru_cache用于存储函数的返回值。可以方便实现记忆化搜索。lru_cache第一个参数为max_size,表示存储的步数,设置为None回退为存储所有情况

如:

from functools import lru_cache
import sys

class Solution:
    def minPathSum(self, grid: List[List[int]]) -> int:
        m, n = len(grid), len(grid[0])

        @lru_cache(None)
        def dfs(i, j):
            if i == m - 1 and j == n - 1:
                return grid[i][j]

            if i == m or j == n:
                return sys.maxsize

            rst = min(dfs(i + 1, j) + grid[i][j], dfs(i, j + 1) + grid[i][j])
            return rst

        return dfs(0, 0)

sys

刷题中常用的两个,一个是获取int最大值,一个是设置最大递归深度。

  • 获取int最大值
import sys
sys.maxsize
  • 设置最大递归深度,Python默认最大递归深度为1000。
sys.setrecursionlimit(1500)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,128评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,316评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,737评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,283评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,384评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,458评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,467评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,251评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,688评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,980评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,155评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,818评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,492评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,142评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,382评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,020评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,044评论 2 352