求一个数组中右边第一个比他大的数(单调栈)

思路:

这个问题,从直觉上来讲,应当是一遍遍历,便可以解决问题。这类问题的特点是,一部分要详尽保留信息,另一部分呈现未解决问题的积累特性。

具体到这个问题,创建一个数据结构保留未解决的子问题,根据特点选择栈。

1、栈里面保留是索引,而非元素,其实这是一个很关键的地方,索引的信息要比内容多,因为可以索引本身就可以确定内容。要牢记这一特点

2、初始栈,里面为第一个元素

3、如果栈不为空,而且当前处理元素比栈顶元素大,则栈顶元素对应的第一个比它大的值,就是该元素

4、弹出栈顶元素,继续处理栈里的元素,直至为空或当前处理元素不大于栈顶元素

5、将当前元素压入栈

6、循环3~5

#include<iostream>
#include<vector>
#include<stack>
using namespace std;
vector<int> get(vector<int> a, int len)
{
	vector<int> res(len, -1);
	vector<int> temp(len);
	int index = 0;
	temp[0] = 0;
	for (int i = 1; i < len; i++)
	{
		while (index>-1 && a[i] > a[(temp[index])])
		{
			res[(temp[index])] = a[i];
			index--;
		}
		index++;
		temp[index] = i;
	}
	return res;
}

vector<int> get2(vector<int> a, int len)
{
	vector<int> res(len, -1);
	stack<int> sta;
	sta.push(0);
	for (int i = 1; i < len; i++)
	{
		while (!sta.empty() && a[i]>a[(sta.top())])
		{
			res[(sta.top())] = a[i];
			sta.pop();
		}
		sta.push(i);
	}
	return res;
}

int main()
{
	vector<int> a = { 2, 5, 3, 7, 1, 2, 8 };
	vector<int> b = get2(a, 7);
	for (int i = 0; i < 7; i++)
	{
		cout << b[i] << " ";
	}
	system("pause");
	return 0;
}
点赞