Leetcode刷题-496下一个更大元素

最新推荐文章于 2024-09-16 06:54:08 发布

weixin_44593040

最新推荐文章于 2024-09-16 06:54:08 发布

阅读量167

点赞数

分类专栏： Leetcode 文章标签： leetcode c++

本文链接：https://blog.csdn.net/weixin_44593040/article/details/124274206

版权

Leetcode 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用C++编程实现一个解决方案，查找给定数组中的每个元素在其后出现的第一个大于它的元素。通过使用栈来存储数组的部分元素，遍历数组两次，实现了先进后出的特点，从而找出每个元素的下一个更大元素。

摘要由CSDN通过智能技术生成

//知识点：栈 stack
//特点：先进后出
#include<iostream>
using namespace std;
#include<stack>
#include<vector>


class Solution {
public:
	vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
		stack<int> st;//存放nums2
		for (auto it : nums2)//相当于for (vector<int>::iterator i = nums2.begin(); i != nums2.end(); i++)
		{
			st.push(it);
		}

		vector<int> num;//返回值

		for (int i = 0; i < nums1.size(); i++)
		{
			stack<int> temp;//临时存放st中出栈的数
			int max = -1;//返回值
			bool isFound = false;//nums1中的值是否在nums2中找到
			while ((!isFound) & (st.size() != 0))
			{
				int m_top = st.top();//栈顶元素
				if (m_top == nums1[i])
				{
					isFound = true;
				}
				else if (m_top > nums1[i])
				{
					max = m_top;
				}
				temp.push(m_top);
				st.pop();
			}
			num.push_back(max);
			while (temp.size() != 0)
			{
				int m_top2 = temp.top();
				temp.pop();
				st.push(m_top2);
			}
		}
		return num;
	}
};

void printVector(vector<int>& v) {

	for (vector<int>::iterator it = v.begin(); it != v.end(); it++) {
		cout << *it << " ";
	}
	cout << endl;
}

void test01() {
	vector<int> v1;
	vector<int> v2;

	v1.push_back(4);
	v1.push_back(1);
	v1.push_back(2);

	v2.push_back(1);
	v2.push_back(3);
	v2.push_back(4);
	v2.push_back(2);

	Solution S;
	vector<int> v3 = S.nextGreaterElement(v1, v2);
	printVector(v3);
}

int main()
{
	test01();
	system("pause");
	return 0;
}