Submission #723923

#TimeUsernameProblemLanguageResultExecution timeMemory
723923adrilenArranging Shoes (IOI19_shoes)C++17
70 / 100
123 ms27680 KiB
#include "shoes.h"
#include<bits/stdc++.h>

using namespace std;
using ll = long long;

constexpr int maxn = 2e5, bits = 18, siz = 1 << bits;

int segtree[siz * 2] = { 0 };

void update(int p)
{
    while (p > 0)
    {
        segtree[p]++;
        p /= 2;
    }
}

int query(int pos, int l, int r, int gl, int gr)
{
    if (l > gr || gl > r) return 0;
    if (gl <= l && r <= gr) return segtree[pos];

    int mid = (l + r) >> 1;
    return query(pos * 2, l, mid, gl, gr) + query(pos * 2 + 1, mid + 1, r, gl, gr);
}


int n;

// Shoe types
set <int> shoes[maxn / 2][2];

ll count_swaps(vector<int> s) {
    n = s.size();

    ll output = 0;

    for (int i = 0; i < n; i++)
    {
        if (s[i] > 0) shoes[s[i]][1].insert(i); // Right
        else shoes[abs(s[i])][0].insert(i);          // Left
    }

    int typ, next_pair;
    for (int i = 0; i < n; i++)
    {
        if (segtree[i + siz]) continue;

        typ = (s[i] > 0);
        shoes[abs(s[i])][typ].erase(i);
        next_pair = *shoes[abs(s[i])][(typ + 1) & 1].begin();
        shoes[abs(s[i])][(typ + 1) & 1].erase(next_pair);


        // cout << i << " " << next_pair << " " << output << " ";
        // cout << query(1, 0, siz - 1, i, next_pair) << "\n";
        output += next_pair - i - query(1, 0, siz - 1, i, next_pair) + typ - 1;

        update(i + siz);
        update(next_pair + siz);
    }

    return output;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...