LeetCode刷题实战379:电话目录管理系统
共 3264字,需浏览 7分钟
·
2021-09-13 15:59
Design a Phone Directory which supports the following operations:
get: Provide a number which is not assigned to anyone.
check: Check if a number is available or not.
release: Recycle or release a number.
示例
// 初始化电话目录,包括 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);
解题
https://www.cnblogs.com/grandyang/p/5735205.html
class PhoneDirectory {
public:
PhoneDirectory(int maxNumbers) {
nums.resize(maxNumbers);
used.resize(maxNumbers);
idx = maxNumbers - 1;
iota(nums.begin(), nums.end(), 0);
}
int get() {
if (idx < 0) return -1;
int num = nums[idx--];
used[num] = 1;
return num;
}
bool check(int number) {
return used[number] == 0;
}
void release(int number) {
if (used[number] == 0) return;
nums[++idx] = number;
used[number] = 0;
}
private:
int idx;
vector<int> nums, used;
};