42 接雨水

42接雨水

42

题解

写完才发现，这题按列求其实也不大啊，我之前没注意n最大2e4。
先说一下我咋写的把，我是按行求的。
复杂度是O（nlgn）。
就是先对数组排序。
从高向低取值。
把值传入优先队列里面（现在发现不需要，就是记录已经取出来的最左和最右两个的位置）
然后每次出现高度变化就计算水量，（宽就是最右减去最左减去已经取出来的个数，高就是高度变化前减去出现高度变化的值。
这样就可以了。
，还有一个小问题，先传一个高度为0的到数组里面，防止最后一个不变化需要特判。

struct stu{
int x;
int hei;
};
bool cmp1(stu a, stu b){
    return a.hei>b.hei;
}

struct tmp2{
    bool operator()(stu a,stu b){ //x从大到小
    return a.x>b.x;
}
};
struct tmp3{
bool operator()(stu a,stu b){ //x从小到大
    return a.x<b.x;
}
};


class Solution {
public:
    vector<stu> ap;
    priority_queue<stu, vector<stu>, tmp2> q2;
    priority_queue<stu, vector<stu> ,tmp3 > q3;
    int trap(vector<int>& height) {
        int ans = 0;
        for(int i=0;i<height.size();i++){
            stu an;
            an.x = i;
            an.hei = height[i];
            ap.push_back(an);
        }
        stu aoo ;
        aoo.hei = 0;
        aoo.x = 0;
        ap.push_back(aoo);
        sort(ap.begin(),ap.end(),cmp1);
        stu nn = ap[0];
        q2.push(nn);
        q3.push(nn);
        int hein = nn.hei;
        int flag = 0;
        int real_hei = nn.hei;
        for(int i=1;i<ap.size();i++){
            nn = ap[i];
            if(nn.hei == hein){
                q2.push(nn);
                q3.push(nn);
                flag = 0;
            }
            else{
                flag = 1;
                stu nnz = q2.top();
                stu nny = q3.top();
               // cout<<"z"<<nnz.x<<" "<<nny.x<<endl;
                int leng = nny.x - nnz.x+1-q2.size();
                int heig = abs(real_hei-ap[i].hei);
                if(leng<0){

                }
                else{
                    ans+= leng*heig;
                    cout<<"leng "<<leng<<" "<<heig<<endl;
                }
                q2.push(nn);
                q3.push(nn);
                hein = nn.hei;
            }
            real_hei = ap[i].hei;
        }
        for(int i=0;i<ap.size();i++){
            cout<<ap[i].hei<<endl;
        }

        return ans;
    }
};

看完题解，用单调栈做只需要O(n);