本文涉及知识点
栈
LeetCode1172. 餐盘栈
我们把无限数量 ∞ 的栈排成一行,按从左到右的次序从 0 开始编号。每个栈的的最大容量 capacity 都相同。
实现一个叫「餐盘」的类 DinnerPlates:
DinnerPlates(int capacity) - 给出栈的最大容量 capacity。
void push(int val) - 将给出的正整数 val 推入 从左往右第一个 没有满的栈。
int pop() - 返回 从右往左第一个 非空栈顶部的值,并将其从栈中删除;如果所有的栈都是空的,请返回 -1。
int popAtStack(int index) - 返回编号 index 的栈顶部的值,并将其从栈中删除;如果编号 index 的栈是空的,请返回 -1。
示例:
输入:
[“DinnerPlates”,“push”,“push”,“push”,“push”,“push”,“popAtStack”,“push”,“push”,“popAtStack”,“popAtStack”,“pop”,“pop”,“pop”,“pop”,“pop”]
[[2],[1],[2],[3],[4],[5],[0],[20],[21],[0],[2],[],[],[],[],[]]
输出:
[null,null,null,null,null,null,2,null,null,20,21,5,4,3,1,-1]
解释:
DinnerPlates D = DinnerPlates(2); // 初始化,栈最大容量 capacity = 2
D.push(1);
D.push(2);
D.push(3);
D.push(4);
D.push(5); // 栈的现状为: 2 4
1 3 5
﹈ ﹈ ﹈
D.popAtStack(0); // 返回 2。栈的现状为: 4
1 3 5
﹈ ﹈ ﹈
D.push(20); // 栈的现状为: 20 4
1 3 5
﹈ ﹈ ﹈
D.push(21); // 栈的现状为: 20 4 21
1 3 5
﹈ ﹈ ﹈
D.popAtStack(0); // 返回 20。栈的现状为: 4 21
1 3 5
﹈ ﹈ ﹈
D.popAtStack(2); // 返回 21。栈的现状为: 4
1 3 5
﹈ ﹈ ﹈
D.pop() // 返回 5。栈的现状为: 4
1 3
﹈ ﹈
D.pop() // 返回 4。栈的现状为: 1 3
﹈ ﹈
D.pop() // 返回 3。栈的现状为: 1
﹈
D.pop() // 返回 1。现在没有栈。
D.pop() // 返回 -1。仍然没有栈。
提示:
1 <= capacity <= 20000
1 <= val <= 20000
0 <= index <= 100000
最多会对 push,pop,和 popAtStack 进行 200000 次调用。
栈
m_stas 记录所有栈的情况。
set m_mNotFull 记录,记录没有满的栈的下标。push 就是 m_mNotFull.begin()。
set m_mNotEmpty记录,出栈就是m_mNotEmpty.rbegin()
这样做会超时,
m_setHasDel 记录删除过的栈,如果删除的栈,又加满了,则从m_setHasDel中删除。
m_iAddPos 从0开始,m_iAddPos号栈满了后,m_iAddPos++。比较两者较小的,入栈。
代码
class DinnerPlates {
public:
DinnerPlates(int capacity):m_iCapacity(capacity){
}
void push(int val) {
if (m_stas[m_iAddPos].size() == m_iCapacity) {
m_iAddPos++;
}
int inx = m_iAddPos;
if (m_setHasDel.size()&&( *m_setHasDel.begin()< inx)) {
inx = *m_setHasDel.begin();
}
m_stas[inx].push(val);
if (m_stas[inx].size() == m_iCapacity) {
m_setHasDel.erase(inx);
}
m_setNotEmpty.emplace(inx);
}
int pop() {
if (m_setNotEmpty.empty()) { return -1; }
return popAtStack(*m_setNotEmpty.rbegin());
}
int popAtStack(int index) {
if (m_stas[index].empty()) { return -1; };
int ret = m_stas[index].top();
m_stas[index].pop();
if (m_stas[index].empty()) {
m_setNotEmpty.erase(index);
}
m_setHasDel.insert(index);
return ret;
}
const int m_iCapacity;
set<int> m_setHasDel, m_setNotEmpty;
stack<int> m_stas[200'000];
int m_iAddPos = 0;
};
扩展阅读
视频课程
有效学习:明确的目标 及时的反馈 拉伸区(难度合适),可以先学简单的课程,请移步CSDN学院,听白银讲师(也就是鄙人)的讲解。
https://edu.csdn.net/course/detail/38771
如何你想快速形成战斗了,为老板分忧,请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176
相关下载
想高屋建瓴的学习算法,请下载《喜缺全书算法册》doc版
https://download.csdn.net/download/he_zhidan/88348653
我想对大家说的话 |
---|
《喜缺全书算法册》以原理、正确性证明、总结为主。 |
闻缺陷则喜是一个美好的愿望,早发现问题,早修改问题,给老板节约钱。 |
子墨子言之:事无终始,无务多业。也就是我们常说的专业的人做专业的事。 |
如果程序是一条龙,那算法就是他的是睛 |
测试环境
操作系统:win7 开发环境: VS2019 C++17
或者 操作系统:win10 开发环境: VS2022 C++17
如无特殊说明,本算法用**C++**实现。