Cod sursa(job #3242198)

Utilizator deerMohanu Dominic deer Data 9 septembrie 2024 19:48:52
Problema Zeap Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.35 kb
#include <bits/stdc++.h>

#define int long long
#define fi first
#define se second

using namespace std;

ifstream fin("zeap.in");
ofstream fout("zeap.out");

void solve_testcase() {
    string s;
    set<int> nr;
    multiset<int> dif;

    auto insert = [&](int x) {
        if (!nr.empty()) {
            auto it = nr.lower_bound(x);
            if (it != nr.begin()) {
                dif.insert(x - *prev(it));  // Difference with the previous element
            }
            if (it != nr.end()) {
                dif.insert(*it - x);  // Difference with the next element
            }
            if (it != nr.begin() && it != nr.end()) {
                dif.erase(dif.find(*it - *prev(it)));  // Remove old difference between neighbors
            }
        }
        nr.insert(x);
    };

    auto erase = [&](int x) {
        if (nr.find(x) != nr.end()) {
            if (nr.size() > 1) {
                auto it = nr.find(x);
                if (it != nr.begin()) {
                    dif.erase(dif.find(x - *prev(it)));  // Remove difference with the previous element
                }
                if (next(it) != nr.end()) {
                    dif.erase(dif.find(*next(it) - x));  // Remove difference with the next element
                }
                if (it != nr.begin() && next(it) != nr.end()) {
                    dif.insert(*next(it) - *prev(it));  // Restore difference between previous and next neighbors
                }
            }
            nr.erase(x);
            return;
        }
        fout << "-1\n";
    };

    auto get_max = [&]() -> int {
        if (nr.size() >= 2)
            return *nr.rbegin() - *nr.begin();
        return -1;
    };

    auto get_min = [&]() -> int {
        if (!dif.empty())
            return *dif.begin();
        return -1;
    };

    auto search = [&](int x) -> int {
        return (nr.find(x) != nr.end());
    };

    while (getline(fin, s)) {
        if (s == "MAX")
            fout << get_max() << '\n';
        if (s == "MIN")
            fout << get_min() << '\n';
        if (s[0] == 'I')
            insert(stoi(s.substr(2)));
        if (s[0] == 'S')
            erase(stoi(s.substr(2)));
        if (s[0] == 'C')
            fout << search(stoi(s.substr(2))) << "\n";
    }
}

signed main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    int t = 1;
    while (t--)
        solve_testcase();
    return 0;
}