来自北大算法课的Leetcode题解：1797. 设计一个验证系统

最新推荐文章于 2026-01-08 21:36:59 发布

原创最新推荐文章于 2026-01-08 21:36:59 发布 · 165 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

来自北大算法课的Leetcode题解专栏收录该内容

247 篇文章

订阅专栏

该代码实现了一个AuthenticationManager类，用于管理具有固定生存期（timeToLive）的tokens。tokens存储在哈希表中，generate方法添加token，renew方法更新token的时间戳，countUnexpiredTokens方法计算并移除过期的tokens。关键在于利用哈希表高效查找和删除过期元素。

代码仓库：Github | Leetcode solutions @doubleZ0108 from Peking University.

解法1(T55% S98%)：没有很复杂，通过一个哈希表来维护即可
- generate()：因为题中条件说token不会重复，因此直接加入哈希表即可
- renew()：首先判断token是否合法（存在且没过期），更新时间
- count()：对于哈希表中每个元素判断是否过期，如果过期则删除否则计数，一般这种一边遍历一遍删除的可能会出问题，所以推荐把所有过期的key先保存在数组里，再单独一次遍历删除会比较稳妥

class AuthenticationManager:

    def __init__(self, timeToLive: int):
        self.timeToLive = timeToLive
        self.table = {}

    def generate(self, tokenId: str, currentTime: int) -> None:
        self.table[tokenId] = currentTime

    def renew(self, tokenId: str, currentTime: int) -> None:
        if tokenId not in self.table:
            return
        if self.table[tokenId]+self.timeToLive <= currentTime:
            return
        self.table[tokenId] = currentTime

    def countUnexpiredTokens(self, currentTime: int) -> int:
        res = 0
        tmp = []
        for key, val in self.table.items():
            if val+self.timeToLive <= currentTime:
                tmp.append(key)
            else:
                res += 1
        for key in tmp:
            self.table.pop(key)
        return res