暴力法:
// 解一:核心思路,按行求,当遇到第一个大于等于当前行的列时,后续遇到比当前行低的列就
// 加1,直到遇到大于等于当前行高的列时就sum一次,每次计算一行,temp需要置为0,status=false
// 时间复杂度:n*Max(int[] height) 空间O(1)
class Solution {
public int trap(int[] height) {
boolean status = false;
int len = height.length;
int min = findMax(height);
int sum = 0,temp = 0;
for(int i=1;i<=min;++i){
for(int j=0;j< len;++j){
if(height[j]>=i){
status = true;
sum = sum + temp;
temp = 0;
}else{
if(status){
temp = temp +1;
}
}
}
temp = 0;
status = false;
}
return sum;
}
public int findMax(int[] height){
int min = 0;
for(int i=0;i< height.length;++i){
if(min<height[i]){
min = height[i];
}
}
return min;
}
}
class Solution {
/*
按列求:
第一列最后一列肯定不会装水
第二列开始,只需要找到左边最高LM,右边最高RM,结果为 Max(LM,RM)-cur 大于0则计入总数
时间复杂度:n*n 空间O(1)
*/
public int trap(int[] height) {
int sum =0;
for(int i=1;i<height.length-1;++i){
int t = Math.min(findMaxLeft(i,height),findMaxRight(i,height)) - height[i];
sum = t>0?sum+t:sum;
}
return sum;
}
int findMaxLeft(int i,int[] height){
int temo = 0;
for(int j=i-1;j>=0;j--){
if(temo<height[j]){
temo = height[j];
}
}
return temo;
}
int findMaxRight(int i,int[] height){
int temo = 0;
for(int j=i+1;j<height.length;j++){
if(temo<height[j]){
temo = height[j];
}
}
return temo;
}
}
动态规划解法:
class Solution {
/*
按列求,由于每列只会用到一次左边右边的最大高度
dp[i] 为当前列左边或右边最大行高;dp[i] = Max(dp[i-1]|dp[i+1],height[i]);
用数组存当前行左边最大高,右边行最大高即可
时间复杂度:n 空间O(n)
*/
public int trap(int[] height) {
if(height.length ==0){
return 0;
}
int sum =0;
int[] maxLeft = new int[height.length];
int[] maxRight = new int[height.length];
maxLeft[0] = height[0];
maxRight[height.length-1] = height[height.length-1];
for(int i=1;i<height.length-1;++i){
maxLeft[i] = Math.max(height[i],maxLeft[i-1]);
}
for(int i= height.length-2;i>0;--i){
maxRight[i] = Math.max(height[i],maxRight[i+1]);
}
for(int i=1;i<height.length-1;++i){
int t = Math.min(maxRight[i],maxLeft[i]) - height[i];
sum = t>0?sum+t:sum;
}
return sum;
}
}