Python | Leetcode Python题解之第432题全O(1)的数据结构

题目:

题解:

python 复制代码
class Node:
    def __init__(self, key="", count=0):
        self.prev = None
        self.next = None
        self.keys = {key}
        self.count = count

    def insert(self, node: 'Node') -> 'Node':  # 在 self 后插入 node
        node.prev = self
        node.next = self.next
        node.prev.next = node
        node.next.prev = node
        return node

    def remove(self):  # 从链表中移除 self
        self.prev.next = self.next
        self.next.prev = self.prev

class AllOne:
    def __init__(self):
        self.root = Node()
        self.root.prev = self.root
        self.root.next = self.root  # 初始化链表哨兵,下面判断节点的 next 若为 self.root,则表示 next 为空(prev 同理)
        self.nodes = {}

    def inc(self, key: str) -> None:
        if key not in self.nodes:  # key 不在链表中
            if self.root.next is self.root or self.root.next.count > 1:
                self.nodes[key] = self.root.insert(Node(key, 1))
            else:
                self.root.next.keys.add(key)
                self.nodes[key] = self.root.next
        else:
            cur = self.nodes[key]
            nxt = cur.next
            if nxt is self.root or nxt.count > cur.count + 1:
                self.nodes[key] = cur.insert(Node(key, cur.count + 1))
            else:
                nxt.keys.add(key)
                self.nodes[key] = nxt
            cur.keys.remove(key)
            if len(cur.keys) == 0:
                cur.remove()

    def dec(self, key: str) -> None:
        cur = self.nodes[key]
        if cur.count == 1:  # key 仅出现一次,将其移出 nodes
            del self.nodes[key]
        else:
            pre = cur.prev
            if pre is self.root or pre.count < cur.count - 1:
                self.nodes[key] = cur.prev.insert(Node(key, cur.count - 1))
            else:
                pre.keys.add(key)
                self.nodes[key] = pre
        cur.keys.remove(key)
        if len(cur.keys) == 0:
            cur.remove()

    def getMaxKey(self) -> str:
        return next(iter(self.root.prev.keys)) if self.root.prev is not self.root else ""

    def getMinKey(self) -> str:
        return next(iter(self.root.next.keys)) if self.root.next is not self.root else ""
相关推荐
EelBarb8 分钟前
python:基于django的html二维码页面生成功能页面
python·django·html
Rverdoser25 分钟前
macOS 系统中python的安装步骤
开发语言·python·macos
亮剑201832 分钟前
文科生学pytorch——一些概念的解释
人工智能·pytorch·python
爱技术的小伙子1 小时前
【30天玩转python】网络爬虫开发
开发语言·爬虫·python
妃衣1 小时前
Tkinter制作登录界面以及登陆后页面切换--用户数据从数据库获取并进行合法性校验(二)
python·sqlite·tkinter
luluvx1 小时前
LeetCode[中等] 78.子集
算法·leetcode·职场和发展
apz_end1 小时前
Python Flask网页开发基本框架
开发语言·后端·python·学习·flask·编程学习
沪上小姐1 小时前
python用两类循环嵌套打印正置九九乘法口诀表和倒置九九乘法口诀表
开发语言·python
Hiweir ·1 小时前
BERT训练之数据集处理(代码实现)
人工智能·python·深度学习·自然语言处理·bert
DdddJMs__1351 小时前
C语言 | Leetcode C语言题解之第433题最小基因变化
c语言·leetcode·题解