503.下一个更大元素II
题目链接:力扣
代码:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
stack<int> st;
vector<int> result(nums.size()*2,-1);
st.push(0);
for(int i = 1; i < nums.size(); i++)
{
while(!st.empty() && nums[i] > nums[st.top()])
{
result[st.top()] = nums[i];
st.pop();
}
st.push(i);
}
for(int i = nums.size(); i < nums.size() * 2; i++)
{
while(!st.empty() && nums[i-nums.size()] > nums[st.top()])
{
result[st.top()] = nums[i-nums.size()];
st.pop();
}
st.push(i-nums.size());
}
vector<int> reback(nums.size(),0);
for(int i = 0; i < nums.size(); i++)
{
reback[i] = result[i];
}
return reback;
}
};
代码:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
stack<int> st;
vector<int> result(nums.size(),-1);
st.push(0);
for(int i = 1; i < 2*nums.size(); i++)
{
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. 接雨水
题目链接:力扣
代码(双指针法):
class Solution {
public:
int trap(vector<int>& height) {
vector<int> leftHight(height.size(),0);
vector<int> rightHight(height.size(),0);
int size = height.size();
leftHight[0] = height[0];
for(int i = 1; i < height.size(); i++)
{
leftHight[i] = max(leftHight[i-1],height[i]);
}
rightHight[size - 1] = height[size-1];
for(int i = size-2; i >= 0; i--)
{
rightHight[i] = max(rightHight[i+1],height[i]);
}
int result = 0;
for(int i = 0; i < size; i++)
{
int count = min(leftHight[i],rightHight[i]) - height[i];
if(count > 0)
result += count;
}
return result;
}
};
这个是按行来的,左边的最大和右边的最大取最小然后减去当前高度 如果不小于 说明是凹进去的 给累加一下
代码(单调栈):
class Solution {
public:
int trap(vector<int>& height) {
stack<int> st;
st.push(0);
int result = 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 if(height[i] > height[st.top()])
{
while(!st.empty() && height[i] > height[st.top()])
{
int mid = st.top();
st.pop();
if(!st.empty())
{
//比较左右两边的高度取小的那个 然后和当前高度求差 算出高度差
int h = min(height[i],height[st.top()]) - height[mid];
int w = i - st.top() - 1;
result += h*w;
}
}
st.push(i);
}
}
return result;
}
};
上面那个是按列求和这个是按行求和 一旦栈顶比当前小,则说明有凹槽,然后找栈顶左右两边的min和栈顶求差,就是栈顶这一部分的高,然后求个宽