代码随想录算法训练营第四十五天| 503.下一个更大元素II 42. 接雨水

503.下一个更大元素II 

要求：

数组是环，需要找到下一个最大的元素

思路1：

先作为直线遍历，然后没有的节点，放到首部，再找比他大的节点

注意：头节点

代码：

// 要求：返回循环数组中下一个更大的数字步数
// 思路：先不循环遍历，
// 然后对每个-1节点，以他为起始，放到数组的开头，计算有几个，仅计算一遍
//
vector<int> nextGreaterElements_complicate(vector<int>& nums) {
    vector<int> result(nums.size(), INT_MIN);
    stack<int> st;

    st.push(0);
    for (int i = 1; i < nums.size(); i++)
    {
        if (nums[i] <= nums[st.top()])
        {
            st.push(i);
        }
        else
        {
            while (!st.empty() && nums[i] > nums[st.top()])
            {
                result[st.top()] = nums[i];
                st.pop();
            }
            st.push(i);
        }
    }

    for (int i = 1; i < nums.size(); i++)
    {
        if (result[i] == INT_MIN)
        {
            cout << result[i] << endl;
            for (int j = 0; j < i; j++)
            {
                if (nums[i] < nums[j])
                {
                    result[i] = nums[j];
                    break;
                }
            }
        }
        if (result[i] == INT_MIN)
        {
            result[i] = -1;
        }
    }

    if(result[0]== INT_MIN)result[0] = -1;

    return result;
}

 

思路2：

模拟遍历两次，也就是重复的把前面的节点，放到后面节点的后面

代码：

// 新思路：模拟走两边，这样就可以把前面的节点放到后面的节点的后边
//
vector<int> nextGreaterElements(vector<int>& nums) {
    vector<int> result(nums.size(),-1);
    stack<int>st;

    st.push(0);
    for (int i = 1; i < nums.size() * 2; i++)
    {
        if (nums[i % nums.size()] <= nums[st.top()])
        {
            st.push(i % nums.size());
        }
        else
        {
            while (!st.empty() && nums[i % nums.size()] > nums[st.top()])
            {
                result[st.top()] = nums[i % nums.size()];
                st.pop();
            }

            st.push(i % nums.size());
        }
    }

    return result;
}

 42. 接雨水  

要求：

给一组数组，然后把雨水放到里面的凹槽中

思路1：

找出左边和右边最低的高度，然后 和 自己的高度相减，就是目前的雨水

注意 ：首尾不算

思路2：

找到右边最高的高度

左右中最低的高度-本身，就是水的高度，然后再*长度，就是体积

代码：

// 要求：当前节点找到最近比他大或等于的节点，以当前节点填充当前节点和最大节点之间的数值
// 公式：min(左侧最高，右侧最奥)-height 
// 注意首尾不接水
// 
// 思路1：
// 当前仅算右侧最高，并没有算左侧最高
// 思路2：
// 横向：
//    小于，放进去，等于，踢出来
//    大于，当前节点就是底部，左边和右边最小值-底部，就是雨水的高度，然后左-右-1，就是长度
//
int trap(vector<int>& height) {
    // 存放当前节点 <= 节点的下标
    int result = 0;
    stack<int> st;

    st.push(0);
    for (int i = 1; i < height.size(); i++)
    {
        if (height[i] < height[st.top()])
        {
            st.push(i);
        }
        else if (height[i] == height[st.top()]) {
            st.pop();
            st.push(i);
        }
        else
        {
            while (!st.empty() && height[i] > height[st.top()])
            {
                int mid = st.top(); st.pop();

                if (!st.empty())
                {
                    int left = st.top();
                    int right = i;

                    int height_ = min(height[right], height[left]) - height[mid];

                    int width = right - left - 1;

                    result += height_ * width;
                }
            }

            st.push(i);
        }

    }


    return result;
}