代码随想录第五十九天

最新推荐文章于 2024-06-16 22:31:26 发布

weixin_47880957

最新推荐文章于 2024-06-16 22:31:26 发布

阅读量122

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/weixin_47880957/article/details/127927603

版权

代码随想录第五十九天

503.下一个更大元素II

给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。
将原数组全部复制一份接在原数组后面，其情况与502一模一样，使用从栈顶到栈底递增的单调栈完成记录，当遍历到的元素值比当前栈顶元素值大时，需要更新result数组
代码如下：

vector<int> nextGreaterNumber(vector<int>& nums)
	{
		vector<int>result(2*nums.size(), -1);
		vector<int>nums1(2 * nums.size());
		for (int i=0;i<2*nums.size();i++)
		{
			nums1[i] = nums[i % nums.size()];
		}
		stack<int>stackRe;
		stackRe.push(0);
		for (int i=1;i<2*nums.size();i++)
		{
			while (!stackRe.empty()&&nums1[i]>nums1[stackRe.top()])
			{
				int top = stackRe.top();
				result[top] = nums1[i];
				stackRe.pop();
			}
			stackRe.push(i);
		}
		vector<int>re(result.begin(),result.begin()+nums.size());
		return re;
	}

42. 接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
使用单调栈时，需要根据当前元素与当前栈顶元素大，相等，小做三种策略的动作，雨水收集在当前元素比栈顶元素大时进行收集
代码如下：

int rainWater(vector<int>& nums)
	{
		int result = 0;
		stack<int>stackWater;
		stackWater.push(0);
		for (int i=1;i<nums.size();i++)
		{
			if (nums[i] == nums[stackWater.top()])
			{
				while (!stackWater.empty()&&nums[i]==nums[stackWater.top()])
				{
					stackWater.pop();
				}
				stackWater.push(i);
			}
			else if (nums[i] < nums[stackWater.top()])
			{
				stackWater.push(i);
			}
			else
			{
				while (!stackWater.empty() && nums[i] > nums[stackWater.top()])
				{
					int top = stackWater.top();
					stackWater.pop();
					if (!stackWater.empty())
					{
						int tmp = (min(nums[i], nums[stackWater.top()]) - nums[top]) * (i - stackWater.top() - 1);
						result += tmp;
					}
				}
				stackWater.push(i);
			}
			
		}
		return result;
	}