379.电话目录管理系统.py

# 设计一个电话目录管理系统，让它支持以下功能：
#
# get: 分配给用户一个未被使用的电话号码，获取失败请返回 -1
# check: 检查指定的电话号码是否被使用
# 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);

class PhoneDirectory:

    def __init__(self, maxNumbers: int):
        """
        Initialize your data structure here
        @param maxNumbers - The maximum numbers that can be stored in the phone directory.
        """
        self.list = [0 for _ in range(maxNumbers)]

    def get(self) -> int:
        """
        Provide a number which is not assigned to anyone.
        @return - Return an available number. Return -1 if none is available.
        """
        for i,x in enumerate(self.list):
            if x == 0:
                self.list[i] = 1
                return i
        return -1

    def check(self, number: int) -> bool:
        """
        Check if a number is available or not.
        """
        return self.list[number] == 0

    def release(self, number: int) -> None:
        """
        Recycle or release a number.
        """
        self.list[number] = 0


# Your PhoneDirectory object will be instantiated and called as such:
# obj = PhoneDirectory(maxNumbers)
# param_1 = obj.get()
# param_2 = obj.check(number)
# obj.release(number)