다음으로 큰 요소는 그 다음의 첫 번째 큰 요소입니다. 예를 들어 보겠습니다.
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