描述
給一個數組,對於每一個元素,找出它之前第一個比它小的元素的值。如果沒有,則輸出它本身。
輸入的數組長度爲n,n <= 100000。
元素的大小爲x,x <= 1000000。
樣例
給出 list = [2,3,6,1,5,5],返回 [2,2,3,1,1,1]。
解釋:
依據題意,找出每個數字前面第一個比它小的元素。
給出 list = [6,3,1,2,5,10,9,15],返回[6,3,1,1,2,5,5,9]。
解釋:
依據題意,找出每個數字前面第一個比它小的元素。
思路
利用棧,構建棧s和數組res,對於元素num[i],如果s的棧頂元素大於num[i],出棧。若棧空,表明num[i]前沒有比它小的元素,num[i]壓棧並添加到res;否則,將棧頂添加到res,並把num[i]壓棧。
#ifndef C1416_H
#define C1416_H
#include<iostream>
#include<vector>
#include<stack>
using namespace std;
class Solution {
public:
/**
* @param num: The arry you should handle
* @return: Return the array
*/
vector<int> getPreviousNumber(vector<int> &num) {
// Write your code here
if (num.empty())
return num;
stack<int> s;
vector<int> res;//返回最終結果
s.push(num[0]);
res.push_back(num[0]);
//對於元素num[i],如果s的棧頂元素大於num[i],出棧
//若棧空,表明num[i]前沒有比它小的元素,num[i]壓棧並添加到res
//否則,將棧頂添加到res,並把num[i]壓棧
for (int i = 1; i < num.size(); ++i)
{
while (!s.empty()&&num[i]<=s.top())
{
s.pop();
}
if (s.empty())
{
s.push(num[i]);
res.push_back(num[i]);
}
else
{
res.push_back(s.top());
s.push(num[i]);
}
}
return res;
}
};
#endif