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