leetcode 42接雨水

leetcode 42题目

思路：
我们要找到每一个位置雨水的高度，这个高度等于该位置左右两边最大值中的小者减去当前位置的高度。所以解决这个问题关键就变成了给定一个位置，如何找到其左右两边的最大值；

方法一：
用两个循环分别找到左边最大值和右边最大值，但是时间复杂度为O(n^2),无法通过测试。

int trap(vector<int>& height)
{
    int ans = 0;
    int size = height.size();
    for (int i = 1; i < size - 1; i++) {
        int max_left = 0, max_right = 0;
        for (int j = i; j >= 0; j--) { //Search the left part for max bar size
            max_left = max(max_left, height[j]);
        }
        for (int j = i; j < size; j++) { //Search the right part for max bar size
            max_right = max(max_right, height[j]);
        }
        ans += min(max_left, max_right) - height[i];
    }
    return ans;
}

方法二：
用两个数组提前保存好每个位置左边和右边的最大值，时间复杂度O(n),空间复杂度也为O(n);

int trap(vector<int>& height) {
    if(height.size()<3)return 0;
    int res=0;
    vector<int>l_max(height),r_max(height);
    int maxl=height[0],maxr=height.back();
    for(int i=1;i<height.size();++i)
    {
        maxl=max(maxl,height[i]);
        l_max[i]=maxl;
        
    }  
    for(int i=height.size()-2;i>-1;--i)
    {
            maxr=max(maxr,height[i]);
            r_max[i]=maxr;
    }
    
    for(int i=1;i<height.size();++i)
    {
        res+=(min(l_max[i],r_max[i])-height[i]);
    }
        return res;
    }

方法三：
我们在遍历数组时维护一个栈。如果当前的条形块小于或等于栈顶的条形块，我们将条形块的索引入栈，意思是当前的条形块被栈中的前一个条形块界定。如果我们发现一个条形块长于栈顶，我们可以确定栈顶的条形块被当前条形块和栈的前一个条形块界定，因此我们可以弹出栈顶元素并且累加答案到 res;

int trap(vector<int>& height) {
    if(height.size()<3)return 0;
    int res=0;
    stack<int>st;
    for(int i=0;i<height.size();++i)
    {
        while(!st.empty()&&height[i]>height[st.top()])
        {
            auto top=st.top();
            st.pop();
            if(st.empty())break;
            int dis=i-st.top()-1;
            int h=min(height[i],height[st.top()])-height[top];
            res+=dis*h;      
        }
        st.push(i);``
    }
        return res;
    }

方法四：
双指针法，时间复杂度O（n）,空间复杂度O（1）；

int trap(vector<int>& height)
{
    int left = 0, right = height.size() - 1;
    int ans = 0;
    int left_max = 0, right_max = 0;
    while (left < right) {
        if (height[left] < height[right]) {
            height[left] >= left_max ? (left_max = height[left]) : ans += (left_max - height[left]);
            ++left;
        }
        else {
            height[right] >= right_max ? (right_max = height[right]) : ans += (right_max - height[right]);
            --right;
        }
    }
    return ans;
}