接雨水C++
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)
#include<iostream>
#include<vector>
using namespace std;
int main()
{
int n; char c; int sum1=0; vector<int>x, a;//x存储数,a存储进栈的下标
int maa;
while(scanf("%d",&n))
{
x.push_back(n);
c = getchar();
if (c == '\n')break;
}
vector<int> result(x);
for(int i=0;i<x.size();++i)
{
while (!a.empty()&&x[a.back()]<x[i])//单减;
{
maa = 0;
n = a.back();//记录位置
a.pop_back();//此时栈顶为上一个比他大的数字位置;
if (!empty(a))
{
maa = min(x[a.back()], x[i]);
for (int j = a.back() + 1; j < i; ++j)x[j] = maa;
}
//if (empty(a))result[n] = i * x[n];
}
a.push_back(i);
}
//for (int i = 0; i < result.size(); ++i)cout << result[i]<<" ";
for (int i = 0; i < result.size(); ++i)sum1+=(x[i]-result[i]);
cout << sum1;
return 0;
}//1 1 2 1 3 0 2 0 5 0 4 0 1 1 2:16
注意对单调栈的理解,即以O(n)的速度得知每个元素的左右各两边第一个比他(小)大的位置上;