《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
#include
#include
#include
#include
#include
#include
using namespace std;
vector findNext(vector A, int n) {
	stack s;
	s.push(-1);
	int sz = A.size();
	vector 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(vectormat, int n)
{
	for (int j = 0; j < n; j++)
	{
		cout << mat[j] << " ";
	}
	cout << endl;
}

int main()
{
	vector 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程序员面试宝典

群) 欢迎你到来哦,看了博文给点脚印呗,谢谢啦~~


你可能感兴趣的:(《Cracking the Coding Interview程序员面试金典》----下一个元素(下一个比他大的))