PTA储水问题(双指针|单调栈)

给定一系列非负整数，将这些数据看作一个柱子高度图，计算按此排列的柱子，下雨之后能接多少雨水。(数组以外的区域高度视为0)

数据范围：0≤n≤106
，数组中每个值满足 0 <val≤109

输入格式:

第一行是n值，第二行是n个非负数

输出格式:

输出能接的雨水单位数

输入样例:

6
3 1 2 5 2 4

输出样例:

5

说明：柱子高度分别为3,1,2,5,2,4, 如图所示。在这种情况下，可以接 5个单位的雨水，蓝色的为雨水 。

AC代码(双指针做法)：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n,x,y;
long long res;
int a[N];
int main()
{
	cin>>n;
	for(int i=0;i<n;i++) cin>>a[i];
	int l=0,r=n-1;
	while(true)
	{
		if(l>=r) break;
		x=a[l];
		y=a[r];
		if(x<y){
			while(l<r&&a[++l]<=x){
				res+=x-a[l];
			}
		}else{
			while(l<r&&a[--r]<=y){
				res+=y-a[r];
			}
		}
	}
	cout<<res;
	return 0;
}

AC代码(单调栈做法)：

解法：使用单调栈(单调递增栈：单调递增栈就是从栈底到栈顶数据是从大到小)。

若栈为空或当前读入元素x小于等于栈顶，x的下标入栈；

若当前读入的元素x大于栈顶下标对应值，出栈给t，此时产生的储水区域为：

以a[t]为水平线，t到栈顶下标为长度，min(x,a[栈顶])为高度构成的(矩形)区域

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int a[1000000];
stack<int>s;
int main()
{
	int n;
	cin>>n;
	ll res=0;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		while(!s.empty()&&a[i]>a[s.top()])
		{
			int t=s.top();
			s.pop();
			if(s.empty()) 
			    break;
			ll h=min(a[i],a[s.top()]);
			res+=(i-s.top()-1)*(h-a[t]);
		}
		s.push(i);
	}
	cout<<res;
	return 0;
} 

 

下面是储水问题的变形题目：

坑坑坑

 

AC代码(双指针做法)：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e6+10;
int n,x,y,cnt,mmax;
int a[N];
signed main()
{
	cin>>n;
	for(int i=0;i<n;i++) cin>>a[i];
	int l=0,r=n-1,f=0;
	while(1){
		if(l>=r) break;
		x=a[l],y=a[r];
		f=0;
		if(x<y){
			while(l<r&&x>=a[++l]){
				if(!f) cnt++,f=1;
				mmax=max(mmax,x-a[l]);
			}
		}else{
			while(l<r&&y>=a[--r]){
				if(!f) cnt++,f=1;
				mmax=max(mmax,y-a[r]);
			}
		}
	}
	cout<<cnt<<endl<<mmax;
	return 0;
}