Submission #597983

#TimeUsernameProblemLanguageResultExecution timeMemory
597983JohannArranging Shoes (IOI19_shoes)C++14
Compilation error
0 ms0 KiB
#include "bits/stdc++.h"
using namespace std;

typedef pair<int,int> pii;
typedef vector<int> vi;
typedef long long ll;
#define sz(x) (int) (x).size()

bool active[200200] = { false };

struct segtree {
    vi arr;
    int size;
    segtree(int _size) {
        size = 1;
        while (size < _size) size *= 2;
        arr.assign(2 * size, 0);
        for (int i = 0; i < _size; ++i) {
            arr[i + size] = i;
        }
    }
    void addOne(int l, int r, int x, int lx, int rx) {
        if (l <= lx && rx <= r) { arr[x] += 1; return; }
        if (r <= lx || rx <= l) { return; }
        int m = (lx + rx) / 2;
        addOne(l, r, 2 * x, lx, m);
        addOne(l, r, 2*x+1, m, rx);
    }
    void addOne(int i) {
        addOne(0, i, 1, 0, size);
    }
    int query(int i, int x, int lx, int rx) {
        if (rx - lx == 1) return arr[x];
        int m = (lx + rx) / 2;
        if (i < m) return query(i, 2 * x, lx, m) + arr[x];
        else return query(i, 2 * x + 1, m, rx) + arr[x];
    }
    int query(int i) {
        return query(i, 1, 0, size);
    }
};

ll count_swaps(vi S) {
    int n = sz(S) / 2;
    ll ans = 0;

    segtree seg(2 * n);
    set<pii> shoes;
    for (int i = 0; i < 2 * n; ++i) shoes.insert({ S[i], i });
    for (int i = 0; i < 2 * n; ++i) {
        if (active[i]) continue;
        active[i] = true;
        int pi = seg.query(i);
        int vi = S[i];
        auto it = shoes.lower_bound({ -vi, 0 });
        int j = it->second;
        active[j] = true;
        int pj = seg.query(j);
        shoes.erase(it);
        shoes.erase(shoes.lower_bound({vi, 0}));
        ans += (pj - pi);
        if (vi < 0) ans -= 1;
        seg.addOne(j);
    }
    return ans;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int n;
    cin >> n;
    vi S(2 * n);
    for (int i = 0; i < 2 * n; ++i) cin >> S[i];

    cout << count_swaps(S) << endl;
}

Compilation message (stderr)

/usr/bin/ld: /tmp/ccgZHCaB.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccrbqxaA.o:shoes.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status