计蒜客 数据结构 栈 模板倒水问题

//单调递减栈
//题目：
//我们来看看这样一道题：地上从左到右竖立着 n 块木板，从 1 到 n 依次编号，（10，5，8，12，6）
//如下图所示。我们知道每块木板的高度，在第 n 块木板右侧竖立着一块高度无限大的木板，
//现对每块木板依次做如下的操作：对于第 i 块木板，我们从其右侧开始倒水，
//直到水的高度等于第 i 块木板的高度，倒入的水会淹没 ai 块木板（如果木板左右两侧水的高度大于等于木板高度即视为木板被淹没），
//求 n 次操作后，所有 ai 的和是多少。
//如图上所示，在第 4 块木板右侧倒水，可以淹没第 5 块和第 6 块一共 2 块木板，a4 = 2。

//例如 n = 5，每块木板高分别为 10，5，8，12，6。sum 记录最后的结果，初始为 0。
//第 1 块木板高 10，栈为空，10 入栈；
//第 2 块木板高 5，栈顶为 10，5 入栈；
//第 3 块木板高 8，栈顶 5 比 8 小，删除栈顶，a2 为 3 - 2 - 	1 = 0，sum = 0。此时栈顶为 10，8 入栈；
//第 4 块木板高 12，栈顶 8 比 12 小，删除栈顶，a 3为 4 - 3 	-1 = 0，sum = 0。此时栈顶为 10 比 12 小，继续删	除栈顶，
//a1 为 4 - 1 - 1 = 2，sum = 2。此时栈为空	，12 入栈。
//第 5 块木板高 6，栈顶为 12，把 6 加入栈中。
//遍历结束后，栈不为空，依次删除栈顶，当前位置为 n + 1，栈顶是第 5 个元素 6，a5 为 6 - 5 - 1 = 0，sum = 2。
//删除栈顶，此时栈顶是第 4 个元素 12，a 4为 6 - 4 - 1 = 1，sum = 3。此时栈为空，得到 sum 为 3，程序结束。

#include<iostream>
#include<cassert>
using namespace std;
class Node {
public:
int id, height;
};
template<class Type> class Stack {
private:
Type *urls;
int max_size, top_index;
public:
Stack(int length_input) {
urls = new Type[length_input];
max_size = length_input;
top_index = -1;
}
~Stack() {
delete[] urls;
}
bool push(const Type &element) {
if (top_index >= max_size - 1) {
return false;
}
top_index++;
urls[top_index] = element;
return true;
}
bool pop() {
if (top_index < 0) {
return false;
}
top_index--;
return true;
}
Type top() {
assert(top_index >= 0);
return urls[top_index];
}
bool empty() {
if (top_index < 0) {
return true;
} else {
return false;
}
}
};
int main() {
//n表示有n块木板，ans用来记录最后的结果
int n, ans = 0;
cin>>n;
//定义node类型的栈stack长度为n
Stack<Node> stack(n);
Node temp;
for(int i = 1; i <= n;i++){
//C++中->左侧一定是指针，.的左侧是实体
cin>>temp.height;
temp.id = i;
while(!stack.empty() && stack.top().height <= temp.height){
ans = ans + i - stack.top().id - 1;
stack.pop();
}
stack.push(temp);
}
while(!stack.empty()){
// stack.pop();
ans = ans + n + 1 - stack.top().id - 1;
stack.pop();
}
cout<<ans<<endl;
return 0;
}