[CTCI] 下一个较大元素

下一个较大元素

题目描述

现在我们有一个int数组，请你找出数组中每个元素的下一个比它大的元素。

给定一个int数组A及数组的大小n，请返回一个int数组，代表每个元素比他大的下一个元素,若不存在则为-1。保证数组中元素均为正整数。

测试样例：

[11,13,10,5,12,21,3],7

返回：[13,21,12,12,21,-1,-1]

单调栈！

class NextElement {
public:
    vector<int> findNext(vector<int> A, int n) {
        // write code here
        vector<int> res(n);
        stack<int> stk;
        for (int i = 0; i < n; ++i) {
            while (!stk.empty() && A[i] > A[stk.top()]) {
                res[stk.top()] = A[i];
                stk.pop();
            } 
            stk.push(i);
        }
        while (!stk.empty()) {
            res[stk.top()] = -1;
            stk.pop();
        }
        return res;
    }
};

 

下一个较大元素II

题目描述

现在有一个数组，请找出数组中每个元素的后面比它大的最小的元素，若不存在则为-1。

给定一个int数组A及数组的大小n，请返回每个元素所求的值组成的数组。保证A中元素为正整数，且n小于等于1000。

测试样例：

[11,13,10,5,12,21,3],7

[12,21,12,12,21,-1,-1]

BST！

class NextElement {
public:
    vector<int> findNext(vector<int> A, int n) {
        // write code here
        vector<int> res(n);
        set<int> st;
        for (int i = n - 1; i >= 0; --i) {
            auto it = st.lower_bound(A[i]);
            if (it == st.end()) {
                res[i] = -1;
            } else if (*it != A[i]) {
                res[i] = *it;
            } else {
                ++it;
                if (it == st.end()) res[i] = -1;
                else res[i] = *it;
            }
            st.insert(A[i]);
        }
        return res;
    }
};