【趣味算法题】在数组中，对于所有元素，找出比当前元素大的下一个元素

在数组中，对于所有元素，找出比当前元素大的下一个元素

意思就是，eg.  数组为 　　　　3 1 2 5 4 6 7

那么我们需要得到的结果应该是  5 2 5 6 6 7 -1

 

解决方法如下：

 

1. 暴力匹配： O (n ^ 2 ) 的效率对所有元素匹配过去，效率非常的低
2. 经过提示， 我想到的一种 O ( nlg n ) 效率的算法

　　只需要对数组扫描一次，我们用一个 Priority_queue 来得到当前最小的元素

　　Prority_queue 存放的数据结构为：

struct sc {
    int key, flag;
    bool operator < (const sc & a) const
    { return a.key < key; }
};

　　key 表示元素的值， flag 表示下标，在赋值的时候有用

 

　　算法思路：

　　遍历一遍，如果当前队列为空，那么 push

　　否则，如果队首的最小元素小于 a[i]

　　　　那么在 答案数组中赋值即得到比当前元素大的下一个元素

　　　　如果队列空，break

 

My source code:

#include <iostream>
#include <cstring>
#include <stack>
#include <string>
#include <ctime>
#include <queue>
#include <algorithm>

using namespace std;

const int MAXN = 1000;

int a[MAXN], b[MAXN];

struct sc {
    int key, flag;
    bool operator < (const sc & a) const
    { return a.key < key; }
};


int main () {
    int i, j, k, u, v, n;
    memset (b, -1, sizeof (b));

    cin >> n;
    for (i = 0; i < n; ++i) cin >> a[i];
    priority_queue <sc> que;
    for (i = 0; i < n; ++i) {
        if (que.empty ()) {
            sc temp;
            temp.key = a[i], temp.flag = i;
            que.push (temp);
            continue;
        }
        for (;;) {
            if (que.empty())    break;  //if pop to empty break;

            if (que.top ().key < a[i]) {
                b[que.top ().flag] = a[i];
                que.pop ();
            } else break;
        }
        sc temp;
        temp.key = a[i];
        temp.flag = i;
        que.push (temp);
    }
    for (i = 0; i < n; ++i) {
        cout << a[i] << ' ';
    }
    cout << endl;
    for (i = 0; i < n; ++i) {
        cout << b[i] << ' ';
    }
    cout << endl;


    return 0;
}