下一个较小的元素是其后第一个较小元素的元素。让我们看一个例子。
arr = [1, 2, 3, 5, 4]
5 的下一个较小元素是 4,元素 1、2 的下一个较小元素是, 3 为 -1,因为它们后面没有更小的元素。
算法
用随机数初始化数组
初始化堆栈。
将第一个元素添加到堆栈中。
迭代遍历数组的元素。
如果栈为空,则将当前元素添加到栈中。
-
当当前元素小于堆栈顶部元素时。
打印顶部元素,并将下一个较小元素作为当前元素。 p>
弹出顶部元素。
将元素添加到堆栈中。
当堆栈不为空时。
将下一个较小元素的元素打印为-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