面试题 17.21. 直方图的水量
给定一个直方图(也称柱状图),假设有人从上面源源不断地倒水,最后直方图能存多少水量?直方图的宽度为 1。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的直方图,在这种情况下,可以接 6 个单位的水(蓝色部分表示水)。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
方案一:朴素解法
/**
* 朴素解法
* 对于每一根柱子来讲能装的水都是 (在本身的柱子比左右俩边的矮时)
* min(左边最的柱子,右边最高的柱子 ) - 本身柱子的高度
*
* @param height
* @return
*/
public int trap(int[] height) {
int n=height.length;//长度, 循环次数
int ans=0;
for(int i=1;i<n-1;i++){ //边缘柱子不会同时有两边侧柱,所以可能接到雨水
int cur =height[i]; // 记录当前位置的高度
int l=Integer.MIN_VALUE;
for(int j=0;j<i;j++) l=Math.max(l,height[j]); //记录左边最高的柱子
if(l<=cur) continue;//本身的柱子比左俩边的高时,跳过
int r=Integer.MIN_VALUE;
for(int k=n-1;k>i;k--) r=Math.max(r,height[k]);//记录右边最高的柱子
if(r<=cur) continue;//本身的柱子比右俩边的高时,跳过
ans+=Math.min(l,r)-cur;
}
return ans;
}
方案二:优化版朴素解法
由于朴素解法每次在比较当前位置的左右柱子高度时会遍历所有的柱子,所以我们在优化的时候就从这个地方入手,就在预处理时就把每个位置的左右最高柱子的高度记录下来,这样时间复杂度可以由o(n2) ——> o(n)
/**
* 朴素解法
* 对于每一根柱子来讲能装的水都是 (在本身的柱子比左右俩边的矮时)
* min(左边最的柱子,右边最高的柱子 ) - 本身柱子的高度
*
* @param height
* @return
*/
public static int trap(int[] height) {
int n=height.length;//长度, 循环次数
int ans=0;
// 由于预处理最值的时候,我们会直接访问到 height[0] 或者 height[n - 1],因此要特判一下
if(n==0)return ans;
int []lm=new int[n]; //记录每个位置,左边最高的柱子(比较时包括本身)
lm[0]=height[0];
for(int i=1;i<n;i++) lm[i]=Math.max(lm[i-1],height[i]);
int []rm=new int[n];//记录每个位置,右边最高的柱子(比较时包括本身)
rm[n-1]=height[n-1];
for(int i=n-2;i>=0;i--) rm[i]=Math.max(rm[i+1],height[i]);
for(int i=1;i<n-1;i++){ //边缘柱子不会同时有两边侧柱,所以可能接到雨水
int cur =height[i]; // 记录当前位置的高度
ans+=Math.min(lm[i],rm[i])-cur;
}
return ans;
}
方案三:单调栈解法(还需加强理解)
class Solution {
public int trap(int[] height) {
int n = height.length;
int ans = 0;
Deque<Integer> d = new ArrayDeque<>();
for (int i = 0; i < n; i++) {
while (!d.isEmpty() && height[i] > height[d.peekLast()]) {
int cur = d.pollLast();
// 如果栈内没有元素,说明当前位置左边没有比其高的柱子,跳过
if (d.isEmpty()) continue;
// 左右位置,并有左右位置得出「宽度」和「高度」
int l = d.peekLast(), r = i;
int w = r - l + 1 - 2;
int h = Math.min(height[l], height[r]) - height[cur];
ans += w * h;
}
d.addLast(i);
}
return ans;
}
}
方案四:面积差解法
/**
*
* 面积差解法
*
* 雨水面积=左边扫描面积+右边扫描面积-矩阵大小-柱子面积
*
* @param height
* @return
*/
public int trap(int[] height) {
int n = height.length;
int ans = 0;
int sum=0;//柱子面积
int max=0;//记录最高柱子高度
for(int i=0;i<n;i++){
int cur=height[i];
sum+=cur;
max=Math.max(max,height[i]);
}
int full=n*max;//整个区域的面积
int lSum=0;//从左边扫过的区域面积
int lMax=0;//从左边扫过时最高的柱子
for(int i=0;i<n;i++){
lMax=Math.max(height[i],lMax);
lSum+=lMax;
}
int rSum=0;//从右边扫过的区域面积
int rMax=0;//从右边扫过时最高的柱子
for(int i=n-1;i>=0;i--){
rMax=Math.max(height[i],rMax);
rSum+=rMax;
}
return lSum+rSum-full-sum;
}
参考 宫水三叶
此文章创于本人学习时的记录,如有错误或更优解还请指出