Submission #530700

#	Time	Username	Problem	Language	Result	Execution time	Memory
530700		c28dnv9q3	Mixture (BOI20_mixture)	C++17	60 / 100	2082 ms	1836 KiB

Mixture

#include <iostream>
#include <bits/stdc++.h>
#define lli long long int
#define RBit _Rb_tree_iterator<pair<const pair<lli, lli>, int>>

using namespace std;

long long int a, b, c;

lli ab(lli n) {
    return n < 0 ? -n : n;
}

lli ggt(lli x, lli y) {
    return x == 0 ? y : ggt(y % x, x);
}

pair<lli, lli> toPoint(lli tX, lli tY, lli tZ) {
    //cout << tX << " " << tY << " " << tZ << "\n";
    lli x, y;
    if (a != 0) {
        x = tY * a - tX * b; y = tZ * a - tX * c;
    } else if (b != 0) {
        x = tX * b - tY * a; y = tZ * b - tY * c;
    } else {
        x = tX * c - tZ * a; y = tY * c - tZ * b;
    }
    lli d = ab(ggt(x, y));
    //cout << x << " " << y << "\n";
    if (d == 0) return {0, 0};
    x /= d;
    y /= d;
    return {x, y};
}

pair<lli, lli> inverse(pair<lli, lli> p) {
    return {- p.first, - p.second};
}

struct cmp {
    bool operator()(const pair<lli, lli>& pA, const pair<lli, lli>& pB) const {
        if (pA.first >= 0 && pB.first < 0) return true;
        if (pB.first >= 0 && pA.first < 0) return false;
        if (pA.first == 0 && pB.first == 0) return pA.second > pB.second;
        return pA.second * pB.first > pB.second * pA.first;
    }
};

/*
ostream& operator << (ostream& io, const pair<lli, lli>& p) {
    io << p.first << " " << p.second;
    return io;
}

ostream& operator << (ostream& io, const pair<pair<lli, lli>, int>& p) {
    io << p.first << " x " << p.second << "\n";
    return io;
}
 */

bool okay(map<pair<lli, lli>, int, cmp> circle) {
    if (circle.empty()) return false;
    RBit it = circle.begin();
    RBit inv = circle.upper_bound(inverse((*it).first));
    if (circle.end() == inv) inv = circle.begin();
    RBit invinv = circle.upper_bound(inverse((*inv).first));
    if (circle.end() == invinv) invinv = circle.begin();
    return it != inv && inv != invinv;
}

int main() {
    //long long int t = chrono::system_clock::now().time_since_epoch().count();

    cin >> a >> b >> c;

    int n = 0, d = 0;
    pair<lli, lli> p;
    vector<pair<lli, lli>> regal;
    map<pair<lli, lli>, int, cmp> circle;

    int q;
    cin >> q;

    string inst;
    int pos;

    long long int x, y, z;

    lli ele, inv;

    for (int i = 0; i < q; ++i) {
        //cout << i << "\n";

        cin >> inst;

        if (inst == "A") {
            cin >> x >> y >> z;
            p = toPoint(x, y, z);

            /*
            cout << "\n";
            cout << x << " " << y << " " << z << "\n";
            cout << p;
            cout << "\n\n";
             */

            /*
            cout << "adding\n";
            cout << p;
            cout << "\n";
             */

            ele = (circle[p]);
            inv = (circle[inverse(p)]);

            if (ele > 0 && inv > 0) d--;

            regal.push_back(p);
            if (p.first == 0 && p.second == 0) n++; else {
                ele++;
            }

            /*
            cout << "dtest\n";
            cout << *circle.end();
            cout << circle[p] << "\n";
            cout << *circle.find(p);
            cout << p << "\n";
            cout << *circle.find(inv);
            cout << inv << "\n";
             */

            if (ele > 0 && inv > 0) d++;

            circle[p] = ele;
        }

        if (inst == "R") {
            cin >> pos;
            p = regal[pos - 1];

            ele = (circle[p]);
            inv = (circle[inverse(p)]);

            if (ele > 0 && inv > 0) d--;

            if (p.first == 0 && p.second == 0) n--; else {
                ele--;
            }

            if (ele > 0 && inv > 0) d++;

            circle[p] = ele;
        }

        if (circle[p] == 0) circle.erase(p);
        if (circle[inverse(p)] == 0) circle.erase(inverse(p));

        /*
        cout << "0 0 x " << n << "\n";
        for (pair<pair<lli, lli>, int> j: circle) cout << j;
        cout << "\n";
         */

        if (n > 0) {
            cout << 1;
        } else {
            if (d > 0) {
                cout << 2;
            } else {
                if (okay(circle)) {
                    cout << 3;
                } else {
                    cout << 0;
                }
            }
        }
        cout << "\n";
    }
}

• Subtask 1: 13 / 13
• Subtask 2: 17 / 17
• Subtask 3: 30 / 30
• Subtask 4: 0 / 40