0379：电话目录管理系统

力扣第 379 题

题目

设计一个电话目录管理系统，让它支持以下功能：

1. get: 分配给用户一个未被使用的电话号码，获取失败请返回 -1
2. check: 检查指定的电话号码是否被使用
3. release: 释放掉一个电话号码，使其能够重新被分配

示例：

// 初始化电话目录，包括 3 个电话号码：0，1 和 2。
PhoneDirectory directory = new PhoneDirectory(3);

// 可以返回任意未分配的号码，这里我们假设它返回 0。
directory.get();

// 假设，函数返回 1。
directory.get();

// 号码 2 未分配，所以返回为 true。
directory.check(2);

// 返回 2，分配后，只剩一个号码未被分配。
directory.get();

// 此时，号码 2 已经被分配，所以返回 false。
directory.check(2);

// 释放号码 2，将该号码变回未分配状态。
directory.release(2);

// 号码 2 现在是未分配状态，所以返回 true。
directory.check(2);

提示：

• 1 <= maxNumbers <= 10^4
• 0 <= number < maxNumbers
• 调用方法的总数处于区间 [0 - 20000] 之内

分析

维护一个哈希表即可。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

class PhoneDirectory:

    def __init__(self, maxNumbers: int):
        self.A = set(range(maxNumbers))

    def get(self) -> int:
        return -1 if not self.A else self.A.pop()

    def check(self, number: int) -> bool:
        return number in self.A

    def release(self, number: int) -> None:
        self.A.add(number)

72 ms