Submission #547309

#TimeUsernameProblemLanguageResultExecution timeMemory
547309Sergio_2357Arranging Shoes (IOI19_shoes)C++17
10 / 100
1 ms1236 KiB
#include "shoes.h"
#include <bits/stdc++.h>

using namespace std;

typedef vector<int> vi;
typedef vector<vi> vii;

const int sizea = 100000;
const int sqrts = 316;
const int sqrtn = (sizea / sqrts + 1);

vi norms(sizea);
vi sqrtarr(sqrtn);

void update(int b, int e, int v)
{
    int idx = 0;
    for (int i = b; i < e; i++) {
        norms[i] += v;
        if (i % sqrts == sqrts - 1) {
            idx = i + sqrts;
            break;
        }
    }
    for (int i = idx; i < e; i += sqrts) {
        sqrtarr[i / sqrts] += v;
        idx = i + 1;
    }
    for (int i = idx; i < e; i++) {
        norms[i] += v;
    }
}

int get(int i)
{
    return norms[i] + sqrtarr[i / sqrts];
}

int gidx(int v, int n)
{
    return v + (n / 2);
}

long long count_swaps(vi s)
{
    int n = s.size();
    iota(norms.begin(), norms.end(), 0);
    vi used(n, 0);
    vii last(n + 2);
    for (int i = 0; i < n; i++)
        last[gidx(s[i], n)].push_back(i);
    for (int i = 0; i < n + 2; i++)
        reverse(last[i].begin(), last[i].end());
    long long c = 0;
    for (int i = 0; i < n; i++) {
        if (used[i])
            continue;
        int r = gidx(-s[i], n);
        int nxt = *prev(last[r].end());
        last[s[i]].pop_back();
        last[r].pop_back();
        used[i] = 1;
        used[nxt] = 1;
        int d = get(nxt) - get(i);
        d--;
        if (s[i] > 0)
            d++;
        c += d;
        update(i + 1, nxt, 1);
    }
    return c;
}
#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...