1 Star 0 Fork 0

wd6/LeetCode-1

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
min-stack.cpp 1.69 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 9年前 . Update min-stack.cpp
// Time: O(n)
// Space: O(1)
class MinStack {
public:
void push(int number) {
if (elements_.empty()) {
elements_.emplace(0);
stack_min_ = number;
} else {
elements_.emplace(static_cast<int64_t>(number) - stack_min_);
if (number < stack_min_) {
stack_min_ = number; // Update min.
}
}
}
void pop() {
auto diff = elements_.top();
elements_.pop();
if (diff < 0) {
stack_min_ -= diff; // Restore previous min.
}
}
int top() {
if (elements_.top() > 0) {
return stack_min_ + elements_.top();
} else {
return stack_min_;
}
}
int getMin() {
return stack_min_;
}
private:
stack<int64_t> elements_;
int stack_min_;
};
// Time: O(n)
// Space: O(n)
class MinStack2 {
public:
void push(int number) {
if (cached_min_with_count_.empty() || cached_min_with_count_.top().first > number) {
cached_min_with_count_.emplace(number, 1);
} else if (cached_min_with_count_.top().first == number) {
++cached_min_with_count_.top().second;
}
elements_.emplace(number);
}
void pop() {
if (cached_min_with_count_.top().first == elements_.top()) {
if (--cached_min_with_count_.top().second == 0) {
cached_min_with_count_.pop();
}
}
elements_.pop();
}
int top() {
return elements_.top();
}
int getMin() {
return cached_min_with_count_.top().first;
}
private:
stack<int> elements_;
stack<pair<int, int>> cached_min_with_count_;
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/wd6/LeetCode-1.git
git@gitee.com:wd6/LeetCode-1.git
wd6
LeetCode-1
LeetCode-1
master

搜索帮助