牛客 小米校招 计算题 单调栈 接雨水

给定n个非负整数表示每个宽度为1的柱子的高度题，计算按此排列的柱子，下雨之后能接多少雨水。

经典的题目 leetcode 42 接雨水


单调栈

• 保持栈内单调递减
• 每次把一个矮的出栈都能够加上一大块面积

如下图

while(~scanf("%lld,", &m)) a[++n] = m;
stack<int> stk; //存下标
int sum = 0;
for(int i=1; i<=n; i++) {
	while(stk.size() && a[stk.top()] < a[i]) {
		int cur = stk.top();
		stk.pop();
		if(stk.empty()) break;
		int lef = stk.top(); //栈顶下标
		int rig = i; //当前i的下标
		int h = min(a[lef], a[rig]) - a[cur]; //求得高度
		sum += h * (rig - lef - 1);
	}
	stk.push(i);
}
printf("%lld\n", sum);