C++中按輸入順序排列的下一個更大元素


下一個更大元素是其後第一個更大的元素。讓我們來看一個例子。

arr = [4, 5, 3, 2, 1]

4 的下一個更大元素是 5,而元素 3、2、1 的下一個更大元素是 -1,因為它們之後沒有更大的元素。

演算法

  • 用隨機數初始化陣列。

  • 初始化一個棧和一個數組。

  • 從陣列的末尾開始迭代。

    • 從棧中移除元素,直到棧為空且棧頂元素小於或等於當前元素。

    • 如果棧為空,則沒有下一個更大元素。因此,將 -1 新增到結果陣列中。

    • 如果棧不為空,則棧頂元素是當前元素的下一個更大元素。將其新增到結果陣列中。

    • 將當前元素壓入棧中。

  • 迭代結果陣列並列印每個元素及其下一個更大元素。

實現

以下是上述演算法在 C++ 中的實現

#include <bits/stdc++.h>
using namespace std;
void nextGreaterElements(int arr[], int n) {
   stack<int> s;
   int result[n];
   for (int i = n - 1; i >= 0; i--) {
      while (!s.empty() && s.top() <= arr[i]) {
         s.pop();
      }
      if (s.empty()) {
         result[i] = -1;
      }else {
         result[i] = s.top();
      }
      s.push(arr[i]);
   }
   for (int i = 0; i < n; i++) {
      cout << arr[i] << " -> " << result[i] << endl;
   }
}
int main() {
   int arr[] = { 1, 2, 3, 4, 5 };
   int n = 5;
   nextGreaterElements(arr, n);
   return 0;
}

輸出

如果你執行上面的程式碼,你將得到以下結果。

1 -> 2
2 -> 3
3 -> 4
4 -> 5
5 -> -1

更新於:2021年10月23日

73 次瀏覽

啟動您的職業生涯

透過完成課程獲得認證

開始
廣告
© . All rights reserved.