Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++의 다음 작은 요소

<시간/>

다음으로 작은 요소는 그 다음의 첫 번째 작은 요소인 요소입니다. 예를 들어 보겠습니다.

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

5에 대한 다음으로 작은 요소는 4이고 요소 1, 2, 3에 대한 다음으로 작은 요소는 -1입니다. 그 뒤에 더 작은 요소가 없기 때문입니다.

알고리즘

  • 난수로 배열 초기화

  • 스택을 초기화합니다.

  • 스택에 첫 번째 요소를 추가합니다.

  • 배열의 요소를 반복합니다.

    • 스택이 비어 있으면 현재 요소를 스택에 추가합니다.

    • 현재 요소가 스택의 맨 위 요소보다 작은 동안.

      • 다음으로 작은 요소를 현재 요소로 사용하여 맨 위 요소를 인쇄합니다.

      • 상단 요소를 팝니다.

    • 스택에 요소를 추가합니다.

  • 스택이 비어 있지 않은 동안.

    • 다음으로 작은 요소가 있는 요소를 -1로 인쇄합니다.

구현

다음은 위의 알고리즘을 C++로 구현한 것입니다.

#include <bits/stdc++.h>
using namespace std;
void nextSmallerElements(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[] = { 5, 4, 3, 2, 1 };
   int n = 5;
   nextSmallerElements(arr, n);
   return 0;
}

출력

위의 코드를 실행하면 다음과 같은 결과를 얻을 수 있습니다.

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