《Cracking the Coding Interview程序员面试金典》----下一个元素（下一个比他大的）

时间限制：3秒  空间限制：32768K  热度指数：740

本题知识点：  栈

 算法知识视频讲解

题目描述

现在我们有一个int数组，请你找出数组中每个元素的下一个比它大的元素。

给定一个int数组A及数组的大小n，请返回一个int数组，代表每个元素比他大的下一个元素,若不存在则为-1。保证数组中元素均为正整数。

测试样例：

[11,13,10,5,12,21,3],7

返回：[13,21,12,12,21,-1,-1]

思路：最右边的那个值肯定没有最大值，所以肯定是-1。初始栈为-1。从后向前计算：

（1）如果当前元素大于栈顶元素，则栈顶元素退出，如果还是大于栈顶元素，继续退出，一直遍历栈到-1或者小于栈顶元素。这个元素就是就是当前值的下一个比较大的元素。

（2）如果当前元素小于栈顶元素，栈顶元素就是当前值的下一个比较大的元素

代码如下：

#include<iostream>
#include<vector>
#include<map>
#include<queue>
#include<algorithm>
#include<set>
#include<stack>
using namespace std;
vector<int> findNext(vector<int> A, int n) {
	stack<int> s;
	s.push(-1);
	int sz = A.size();
	vector<int> res(sz);
	int top;
	for (int i = sz - 1; i >= 0; --i){
		top = s.top();
		while (top != -1 && A[i] >= top){
			s.pop();
			top = s.top();
		}
		s.push(A[i]);
		res[i] = top;
	}
	return res;
}

void printVector(vector<int>mat, int n)
{
	for (int j = 0; j < n; j++)
	{
		cout << mat[j] << " ";
	}
	cout << endl;
}

int main()
{
	vector<int> v;
	int m;
	int temp;
	while (cin >> m)
	{
		v.clear();
		for (int j = 0; j < m; j++)
		{
			cin >> temp;
			v.push_back(temp);
		}
		printVector(findNext(v, m), m);
	}
	return 0;
}

不懂的可以加我的QQ群：261035036（IT程序员面试宝典

群） 欢迎你的到来哦，看了博文给点脚印呗，谢谢啦~~