O(n) 取得数组中每个元素右边最后一个比它大的元素

题目

2019.9.7，icpc徐州网络赛的E题 XKC's basketball team ，计蒜客上还可以做。

链接：https://nanti.jisuanke.com/t/41387

Input

The first line contains two integers n and m ( 2 ≤ n ≤ 5∗10^5 ,0 ≤ m ≤ 10^9) .

The following  line contain n integers W1​..Wn​(0 ≤ Wi​ ≤ 10^9) .

Output

A row of n integers separated by spaces , representing the anger of every member .

样例输入

6 13 4 5 6 2 10

样例输出

4 3 2 1 0 -1

题意：

对于数组中的每个元素来说，如果它加上m之后，右边仍然有比它大的数（有多个的话，取距离最远的），这个距离就是我们要的结果，不存在取 -1。所以问题就是，取得数组中每个元素右边最后一个比它大的元素。

思路：

看了 “O(n) 取得数组中每个元素右边第一个比它大的元素” 的思路想到的。

• 建立一个Node (value, index)；value是原数组的值 + m，index是对应的索引。（具体的value是什么，视问题而定）
• 建立壹 Node 的 value 值升序的优先级队列。也就是小根堆。
• i = n - 1，开始向前遍历

不断用当前堆顶的value值与arr[i]进行比较，如果堆顶元素小于arr[i]，则找到需要找的元素。因为堆顶是最小的元素，如果堆顶大于arr[i]，则i--，继续重复当前步骤。

• 直到遍历完成，堆中大概率还会剩下几个（加+m之后，原数组中没有元素比它们大了），这些元素对应的 res 都是 -1 。

 #include <iostream>
 #include <queue>
 #include <stdio.h>
 using namespace std;

 int arr[];
 int anger[];

 struct Node {
     int value, index;
     Node(int v, int i) :value(v), index(i) {};
     bool operator< (const Node& n)const {
         return n.value < value;
     }
 };
 priority_queue <Node>pq;

 int main() {
     int n, m;
     cin >> n >> m;
     for (int i = ; i < n; i++) {
         scanf("%d", &arr[i]);
         anger[i] = -;
         pq.push(Node(arr[i] + m, i));
     }

     for (int i = n - ; i >= ; i--) {
         while (!pq.empty()) {
             int top = pq.top().value;
             if (arr[i] >= top) {
                 int index = pq.top().index;
                 pq.pop();
                 if (i > index) {
                     anger[index] = i - index - ;
                 }
                 else anger[index] = -;
             }
             else break;
         }
     }

     for (int i = ; i < n; i++) {
         printf("%d", anger[i]);
         if (i != n - )printf(" ");
     }

     return ;
 }