时间限制: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程序员面试宝典
群) 欢迎你的到来哦,看了博文给点脚印呗,谢谢啦~~