在 C++ 中为数组中的每一个元素寻找最近的更小值


我们将在本文中了解如何查找数组中每个元素最近的值。如果元素 x 拥有大于它且也存在于数组中的下一个元素,则该元素将成为该元素的较大值。如果元素不存在,则返回 -1。假设数组元素为 [10, 5, 11, 6, 20, 12],则较大的元素为 [11, 6, 12, 10, -1, 20]。由于 20 在数组中没有较大值,则打印 -1。

为解决此问题,我们将使用 C++ STL 中的设置。使用二叉树方法实现集合。在二叉树中,中序后继始终是下一个较大元素。因此,我们可以在 O(log n) 时间内获取元素。

示例

 实时演示

#include<iostream>
#include<set>
using namespace std;
void nearestGreatest(int arr[], int n) {
   set<int> tempSet;
   for (int i = 0; i < n; i++)
      tempSet.insert(arr[i]);
   for (int i = 0; i < n; i++) {
      auto next_greater = tempSet.upper_bound(arr[i]);
      if (next_greater == tempSet.end())
         cout << -1 << " ";
      else
         cout << *next_greater << " ";
   }
}
int main() {
   int arr[] = {10, 5, 11, 6, 20, 12};
   int n = sizeof(arr) / sizeof(arr[0]);
   nearestGreatest(arr, n);
}

输出

11 6 12 10 -1 20

更新日期: 19-12-2019

137 次浏览

开启你的 职业生涯

完成课程获得认证

开始
广告