Submission #722873

#TimeUsernameProblemLanguageResultExecution timeMemory
722873GrandTiger1729Arranging Shoes (IOI19_shoes)C++17
100 / 100
77 ms15164 KiB
#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;

struct BIT {
    int n;
    vector<int> bit;
    BIT(int _n): n(_n){
        bit.resize(n + 1);
    }
    void add(int i, int u){
        for (; i <= n; i += i & -i)
            bit[i] += u;
    }
    int query(int i){
        int ret = 0;
        for (; i > 0; i -= i & -i)
            ret += bit[i];
        return ret;
    }
    int query(int l, int r){
        return query(r) - query(l);
    }
};

long long count_swaps(vector<int> s){
	auto a = s;
    int N = a.size(), n = N / 2;
    vector<vector<int>> pos(N + 1);
    for (int i = N - 1; i >= 0; i--)
        pos[n + a[i]].push_back(i);
    vector<int> idx(N, -1);
    int cnt = 0;
    for (int i = 0; i < N; i++){
        if (idx[i] != -1) continue;
        int j = i, k = pos[n - a[i]].back();
        pos[n + a[i]].pop_back();
        pos[n - a[i]].pop_back();
        if (a[j] > 0) swap(j, k);
        idx[j] = 2 * cnt, idx[k] = 2 * cnt + 1;
        cnt++;
    }
    long long ret = 0;
    BIT bit(N);
    for (int i = 0; i < N; i++){
        ret += bit.query(idx[i], N);
        bit.add(idx[i] + 1, 1);
    }
    return ret;
}
#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...