2021-11-03 每日一题延展-接雨水姊妹题

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;
    }
};