《Cracking the Coding Interview程式設計師面試金典》----下一個元素（下一個比他大的）

塵封的記憶0發表於2017-05-03
原文網址 : https://blog.csdn.net/wxq_wuxingquan/article/details/71154167
View程式設計師面試
時間限制：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程式設計師面試寶典


群） 歡迎你的到來哦，看了博文給點腳印唄，謝謝啦~~





            




相關文章