愚蠢如我
这两个其实还不太一样的
和这个http://acm.hdu.edu.cn/showproblem.php?pid=1506 更不一样(后者是单调栈)
https://www.lintcode.com/problem/trapping-rain-water/description
由于图是酱婶的,所以需要一位一位的算
class Solution {
public:
/**
* @param heights: a list of integers
* @return: a integer
*/
int trapRainWater(vector<int> &heights) {
// write your code here
int len = heights.size();
if (len <= 2) {
return 0;
}
int l = 1;
int r = len - 2;
int tot = 0;
int lmax = heights[0];
int rmax = heights[r + 1];
while (l <= r) {
if (lmax < rmax) {
if (lmax < heights[l]){
lmax = heights[l];
} else {
tot += (lmax - heights[l]);
}
l ++;
} else {
if (rmax < heights[r]) {
rmax = heights[r];
} else {
tot += (rmax - heights[r]);
}
r--;
}
}
return tot;
}
};
事实证明我昨天晚上睡觉前想的 除了空间复杂度高 也不是不能A
class Solution {
public:
/**
* @param heights: a list of integers
* @return: a integer
*/
int trapRainWater(vector<int> &heights) {
// write your code here
if (heights.size() == 0 || heights.size() == 1) {
return 0;
}
int len = heights.size();
vector<int> v1(heights);
vector<int> v2(heights);
int maxn = heights[0];
v1[0] = 0;
for (int i = 1; i < len; i ++) {
v1[i] = maxn;
if (maxn < heights[i])
maxn = heights[i];
}
for (int i = 0; i < len; i ++)
printf("i=%d,v1[i]=%d\n", i, v1[i]);
v2[len - 1] = 0;
maxn = heights[len - 1];
for (int i = len - 2; i >= 0; i --) {
v2[i] = maxn;
if (maxn < heights[i])
maxn = heights[i];
}
for (int i = 0; i < len; i ++)
printf("i=%d,v2[i]=%d\n", i, v2[i]);
int tot = 0;
for (int i = 1; i < len - 1;i ++) {
tot += max(0, min(v1[i], v2[i]) - heights[i]);
printf("i= %d, tot = %d\n", i, max(0, min(v1[i], v2[i]) - heights[i]));
}
return tot;
}
};
回来看leetcode
图是这样以为着 是找两个隔板间的矩形
相同的地方都是两个指针选小的往中间走
class Solution {
public:
int maxArea(vector<int>& height) {
int len = height.size();
if (len <= 1) {
return 0;
}
int l = 0;
int r = len - 1;
int tot = 0;
while (l < r) {
tot = max(tot, (r-l)*min(height[r], height[l]));
if (height[r] < height[l]) {
r--;
} else {
l++;
}
}
return tot;
}
};