Computer >> Máy Tính >  >> Lập trình >> C ++

Phần tử lớn hơn tiếp theo trong C ++

Phần tử lớn hơn tiếp theo là phần tử đầu tiên là phần tử lớn hơn sau nó. Hãy xem một ví dụ.

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

Phần tử lớn hơn tiếp theo cho 4 là 5 và phần tử lớn hơn tiếp theo cho các phần tử 3, 2, 1 là -1 vì không có phần tử nào lớn hơn sau chúng.

Thuật toán

  • Khởi tạo mảng với các số ngẫu nhiên.

  • Khởi tạo một ngăn xếp.

  • Thêm phần tử đầu tiên vào ngăn xếp.

  • Lặp lại qua phần tử của mảng.

    • Nếu ngăn xếp trống, hãy thêm phần tử hiện tại vào ngăn xếp.

    • Trong khi phần tử hiện tại lớn hơn phần tử trên cùng của ngăn xếp.

      • In phần tử trên cùng với phần tử lớn hơn tiếp theo làm phần tử hiện tại.

      • Bật phần tử trên cùng.

    • Thêm phần tử vào ngăn xếp.

  • Trong khi ngăn xếp không trống.

    • In các phần tử có phần tử lớn hơn tiếp theo là -1.

Thực hiện

Sau đây là cách thực hiện thuật toán trên trong C ++

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

Đầu ra

Nếu bạn chạy đoạn mã trên, thì bạn sẽ nhận được kết quả sau.

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