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; 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; }
输出
如果你运行以上代码,你会得到以下结果。
1 -> 2 2 -> 3 3 -> 4 4 -> 5 5 -> -1
广告