该题与leetcode中接雨水,同问题。
以下,用双向队列,单调队列完成此题。(本题是单调减小)
困难之处在于如何来维护这个单调队列。观察题意。有以下解题思路(字丑勿喷)
class Solution {
public:
/**
* max water
* @param arr int整型vector the array
* @return long长整型
*/
long long maxWater(vector<int>& arr) {
// write code here
deque<long long> my_deque;
long long ans = 0;
//维护一个单调队列
for(int i = 0;i<arr.size();i++)
{
if(my_deque.empty())my_deque.push_back((long long) arr[i]); // 为空时压入 只会在一开始的时候出现
else{ // 不为空情况
if(my_deque.back()>=arr[i]) //比尾元素小的时候,入栈
{
my_deque.push_back(static_cast<long long> (arr[i])); //严格递减
}else if(my_deque.back()<arr[i]) //大的时候
{
if(my_deque.front()>arr[i])//与队列首元素做比较
{
int zhi = 1;
while(arr[i]>my_deque.back())
{
zhi++;
ans += (static_cast<long long>(arr[i]) - my_deque.back());
my_deque.pop_back(); //弹出
}
for(int j = 0;j<zhi;j++)
{
my_deque.push_back((long long)arr[i]);
} //补全
}
else if(my_deque.front()<=arr[i]) //比首元素大的情况
{
long long num = my_deque.front();
my_deque.pop_front();//弹出此时的头元素
while(!my_deque.empty()) //此时单调队列会全部排空
{
ans += ( static_cast<long long>(num)-my_deque.front());
my_deque.pop_front();//弹出
}
my_deque.push_back(static_cast<long long> (arr[i])); //压入该元素
}
}
}
}
return ans;
}
};
//2021.4.14 单调栈+双向队列