算法题解—双指针—储水量

描述

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 1。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的直方图，在这种情况下，可以接 6 个单位的水（蓝色部分表示水）。

输入

输入一个正整数N表示数组的大小
再输入一行正整数

输出

输出一个数字表示这个直方图的储水量

题解

朴素做法：
求总的储书量就是求x轴上每一点能储存多少水量之和
那么对于每一点，可以求出其左半边中最长的那一个直方柱和右半边最长的直方柱，两个直方柱取最小值，那么该点与其最小值的差值就是该点能储存的水量。这个类似于木桶原理，就是能储存的水取决于最短的那一块。
容易知道，如果使用朴素做法，时间复杂度会达到O(n^2)级别，对于大数据量必定超时

双指针做法:
根据朴素做法，每一点需要的都是其左半边和右半边中最长的直方柱，那么我们可以先提前处理好每一点左半边和右半边最长直方柱，开两个数组，l[N],r[N] 分别表示在i点时，左右最长直方柱的高度。
那么最后只需要再对每一点求 min(l[i],r[i]) - height[i] 即可
处理两个数组的过程可以采用双指针的算法
时间复杂度可以降到O(n)

代码

#include <iostream>
#include <unordered_map>
#include <vector>


using namespace std;

const int N = 100010;
int n;
int a[N],l[N],r[N];

int main()
{
	cin >> n;
	
	for(int i = 0;i < n;i ++) cin >> a[i];

	l[0] = a[0],r[n - 1] = a[n - 1];
	for(int i = 1;i < n;i ++){
		l[i] = max(l[i - 1],a[i]);
		r[n - i - 1] = max(r[n - i],a[n - i - 1]);
	}	
	
	int res = 0;
	for(int i = 0;i < n;i ++){
		res += min(l[i],r[i]) - a[i];
	}
	
	cout << res << endl;
	return 0;
}