Submission #857583

#TimeUsernameProblemLanguageResultExecution timeMemory
857583sleepntsheepArranging Shoes (IOI19_shoes)C++17
10 / 100
2 ms4700 KiB
#include <vector>
#include <cstdint>
#include <cassert>
#include <iostream>
#include <algorithm>

#define N 200005

int fw[N], posraw[N<<1];
int *pos = posraw+N;

void upd(int p, int k) { for (++p; p < N; p+=p&-p) fw[p] += k; }
int qry(int p) { if(p<0)return 0;int z = 0; for (++p; p; p-=p&-p) z += fw[p]; return z; }

int64_t count_swaps(std::vector<int> v)
{
    int64_t z = 0;
    int n = v.size() >> 1, f=0;

    for (int i = 0; i < (int)v.size(); ++i) pos[v[i]] = i;
    for (int i = 0; i < 2 * n; ++i)
    {
        if (qry(i) - qry(i-1) == 1) {continue;}
        int j = i + (qry(2*n) - qry(i));
        int corresponding = pos[-v[i]] + (qry(2*n) - qry(pos[-v[i]]));
        if (v[i] > 0)
            z += corresponding - j;
        else 
            z += corresponding - j - 1;
        upd(pos[-v[i]], 1);
        ++f;
    }
    assert(qry(2*n)==n);

    return z;
}

#if 0
int main()
{
    std::cout << count_swaps({1,2,-2,-1});
}
#endif

#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...