前一個數 - LintCode

描述
給一個數組,對於每一個元素,找出它之前第一個比它小的元素的值。如果沒有,則輸出它本身。

輸入的數組長度爲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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章