LeetCode42-接雨水I
解题思路:对于每个点接雨水的量都取决于两侧最高柱子的高度,我们至于要两次遍历,求出每个点对应左右侧最高的柱子保存起来,就可以通过这两个数组直接求出各点接水量。
class Solution {
public://动态规划
int trap(vector<int>& height) {//对于每个点保存它左右的最大值,遍历数组计算每个点储水量
int n=height.size(),res=0;
vector<int> LH(n),RH(n);
LH[0]=height[0];RH[n-1]=height[n-1];
for(int i=n-2;i>=0;i--)
{
RH[i]=max(RH[i+1],height[i]);
}
for(int i=1;i<n;i++)
{
LH[i]=max(LH[i-1],height[i]);
res+=min(RH[i],LH[i])-height[i]>0?min(RH[i],LH[i])-height[i]:0;
}
return res;
}
};
优化解法:在上一种解法中我们使用两个数组来存储,为节省空间,我们可以使用双指针的方式从两侧向中间压缩,同时使用两个变量分别保存当前左右侧最高柱子的高度。同时也节省了时间,因为只需要依次遍历。
class Solution {
public:
int trap(vector<int>& height) {
int n=height.size(),res=0;
vector<int> LH(n),RH(n);
int left=0,leftH=height[0],right=n-1,rightH=height[n-1];
while(left<right)
{
leftH=max(height[left],leftH);
rightH=max(height[right],rightH);
if(leftH<rightH)
{
res+=leftH-height[left];
left++;
}
else
{
res+=rightH-height[right];
right--;
}
}
return res;
}
};
LeetCode407-接雨水II
编程实现源自官方题解
这次是直接在官解上加注释了,因为开始思路有了但是一直写不出来T-T
解题思路:核心思路就是木桶原理,即桶内水的高度有最低木板的高度决定;
先把最外围的一圈作为木板,每次选择当前最低的木板,如果这个木板的邻节点都比它大,此木板可删除,邻节点作为新的木板加入;如果邻节点比它小, 那么邻节点可储蓄的水为最低木板高度与自身之差,此时在邻节点的位置设置木板,高度为当前最低木板高度。
编程时通过二元组记录当前位置外围木板的高度以及当前位置;位置信息由二维数组索引转换为一维数组索引保存。
/*先把最外围的一圈作为木板,每次选择当前最低的木板,
如果这个木板的邻节点都比它大,此木板可删除,邻节点作为新的木板加入;
如果邻节点比它小, 那么邻节点可储蓄的水为最低木板高度与自身之差,
此时在邻节点的位置设置木板,高度为当前最低木板高度。*/
typedef pair<int,int> pii;//木板高度-木板位置(位置信息用二维转一位的索引方式记录)
class Solution {
public:
int trapRainWater(vector<vector<int>>& heightMap) {
if (heightMap.size() <= 2 || heightMap[0].size() <= 2) {
return 0;
}
int m = heightMap.size();
int n = heightMap[0].size();
priority_queue<pii, vector<pii>, greater<pii>> pq;
vector<vector<bool>> visit(m, vector<bool>(n, false));//用于记录当前位置是否访问过
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (i == 0 || i == m - 1 || j == 0 || j == n - 1) {
pq.push({heightMap[i][j], i * n + j});
visit[i][j] = true;//标记被访问过
}
}
}
int res = 0;
int dirs[] = {-1, 0, 1, 0, -1};
while (!pq.empty()) {
pii curr = pq.top();
pq.pop();
for (int k = 0; k < 4; ++k) {//上下左右四个点
int nx = curr.second / n + dirs[k];
int ny = curr.second % n + dirs[k + 1];
if( nx >= 0 && nx < m && ny >= 0 && ny < n && !visit[nx][ny]) {//在有效范围内且没有被访问过
if (heightMap[nx][ny] < curr.first) {
res += curr.first - heightMap[nx][ny]; //小于最低木板
}
visit[nx][ny] = true;//更新节点状态
pq.push({max(heightMap[nx][ny], curr.first), nx * n + ny});//二者中选择相对高的木板加入,使用当前访问节点的位置
}
}
}
return res;
}
};