思路:
这个问题,从直觉上来讲,应当是一遍遍历,便可以解决问题。这类问题的特点是,一部分要详尽保留信息,另一部分呈现未解决问题的积累特性。
具体到这个问题,创建一个数据结构保留未解决的子问题,根据特点选择栈。
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;
}