UVA501 - Black Box(vector + lower_bound)

UVA501 - Black Box(vector + lower_bound)

题目链接

题目大意：给你两种操作：Add num，将一个数加入黑盒子中，原本的黑盒子是空的。并且X = 0.Get
就是把X自增1，然后取出第X小的数。题目给的u序列的意思是在第Ui个数加入黑盒子后然后执行get操作。

解题思路：用vector来存放，这样能够实现随机取出第i小的数。然后用lower_bound函数来查找出比这个数大于等于的第一个数的位置，将这个数插入这个位置，复杂度n∗log(n),再用insert插入到vector的对应的位置。

代码：

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;
typedef long long ll;

const int maxn = 3e4 + 5;

vector<ll> v;
vector<ll>::iterator it;
int n, m;
ll num[maxn];
int op;

int main () {

    int T;
    scanf ("%d", &T);
    while (T--) {

        scanf ("%d%d", &n, &m);
        for (int i = 0; i < n; i++)
            scanf ("%lld", &num[i]);
        v.clear();
        int dex = 0;
        for (int i = 0; i < m; i++) {
            scanf ("%d", &op);
            while (dex < op) {

                it = lower_bound(v.begin(), v.end(), num[dex]);
                if (it != v.end())
                    v.insert(it, num[dex++]);
                else
                    v.push_back(num[dex++]);
            }
            printf ("%lld\n", v[i]);
        }

        if (T)
            printf ("\n");
    }
    return 0;
}