vector如何实现有序数组？

rag_doll

于 2024-02-07 09:27:41 发布

阅读量417

点赞数 3

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/acrux1985/article/details/136065275

版权

算法专栏收录该内容

30 篇文章 1 订阅

订阅专栏

文章介绍了如何使用C++的vector和lower_bound函数实现有序数组，以支持O(logn)的时间复杂度进行插入操作，同时应用到求解中位数问题，避免了传统方法中维护最大堆和最小堆的复杂性。

摘要由CSDN通过智能技术生成

有序数组维护一个序列 $a_1,a_2,...,a_n$ ，使得其保持有序，而且支持插入时间为 $O (l o g n)$ 的操作。python中有第三方库SortedList，C++并没有这样的stl库。
但我们可以使用vector和lower_bound进行类似的操作：使用lower_bound找到插入的位置，使用vector::insert插入该位置。

如题洛谷P1168 中位数
正统的做法是维护一个最大堆和一个最小堆，最小堆的数在后半部分（比最大堆大），最大堆的数在前半部分，每次取最大堆的堆顶。但是写起来比较麻烦。
用有序数组可以这么写：

#define _CRT_SECURE_NO_WARNINGS

#include <iostream>
#include <string>
#include <cstring>
#include <climits>
#include <cstdlib>
#include <cmath>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <unordered_map>
#include <algorithm>
#define LT(x) (x * 2)
#define RT(x) (x * 2 + 1)

using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;

int n;
int a[100002];
vi b;   // 有序数组


int main() {
    //freopen("in.txt", "r", stdin);
    scanf("%d", &n);
    for (int i = 0; i < n; ++i) {
        scanf("%d", &a[i]);
    }
    int m = (n + 1) / 2;
    b.push_back(a[0]);
    printf("%d\n", a[0]);
    for (int i = 1; i < m; ++i) {
        int pos = lower_bound(b.begin(), b.end(), a[i * 2 - 1]) - b.begin();
        b.insert(b.begin() + pos, a[i * 2 - 1]);
        pos = lower_bound(b.begin(), b.end(), a[i * 2]) - b.begin();
        b.insert(b.begin() + pos, a[i * 2]);
        printf("%d\n", b[i]);
    }
    return 0;
}