[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 = ; 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()] = -;
             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 - ; i >= ; --i) {
             auto it = st.lower_bound(A[i]);
             if (it == st.end()) {
                 res[i] = -;
             } else if (*it != A[i]) {
                 res[i] = *it;
             } else {
                 ++it;
                 if (it == st.end()) res[i] = -;
                 else res[i] = *it;
             }
             st.insert(A[i]);
         }
         return res;
     }
 };