每日一题 - 240117 - B. Bus of Characters

最新推荐文章于 2025-11-25 11:51:32 发布

原创最新推荐文章于 2025-11-25 11:51:32 发布 · 622 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#算法

题单 - STL超级作业专栏收录该内容

13 篇文章

订阅专栏

本文介绍了如何在C++中使用STL的优先队列数据结构，并展示了如何通过重载运算符实现自定义比较函数，以解决特定问题。时间复杂度分析为O(N*logN)。

B. Bus of Characters

TAG - $算法 - 【 ST L - 优先队列、重载运算符】$
时间复杂度 - $\ast \log N)$

//
#include <bits/stdc++.h>
using namespace std;
// #define int long long

const int N = 2e5 + 6;
struct A {
    int idx, w;
} in[N];

struct cmp0 {
    bool operator()(const A& x, const A& y) const { return y.w < x.w; }
};

struct cmp1 {
    bool operator()(const A& x, const A& y) const { return y.w > x.w; }
};

void solve() {
    int n;
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        int ai;
        scanf("%d", &ai);
        in[i] = (A){i, ai};
    }

    priority_queue<A, vector<A>, cmp0> q0;
    priority_queue<A, vector<A>, cmp1> q1;

    for (int i = 1; i <= n; i++) q0.push(in[i]);

    string s;
    cin >> s;
    for (int i = 0; i < s.size(); i++) {
        if (s[i] == '0') {
            A tt = q0.top(); q0.pop();

            printf("%d%c", tt.idx, " \n"[i == s.size() - 1]);
            q1.push(tt);
        } else if (s[i] == '1') {
            A tt = q1.top(); q1.pop();

            printf("%d%c", tt.idx, " \n"[i == s.size() - 1]);
        }
    }
}

signed main() {
    int t = 1;
    // scanf("%d", &t);
    while (t--) solve();

    return 0;
}