Submission #469572

#TimeUsernameProblemLanguageResultExecution timeMemory
469572spike1236Arranging Shoes (IOI19_shoes)C++14
10 / 100
1096 ms332 KiB
#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
#define f first
#define s second
#define ll long long
#define ld long double
#define all(_v) _v.begin(), _v.end()
#define sz(_v) (int)_v.size()
#define pii pair <int, int>
#define pll pair <ll, ll>
#define veci vector <int>
#define vecll vector <ll>

const int dx[4] = {1, -1, 0, 0};
const int dy[4] = {0, 0, -1, 1};
const double PI = 3.1415926535897932384626433832795;
const double eps = 1e-9;
const int MOD1 = 1e9 + 7;
const int MOD2 = 998244353;

const int MAXN = 2e5 + 10;
int t[MAXN], n;
int pos[MAXN][2];

void upd(int r, int val) {
    for(; r < MAXN; r += r & -r) t[r] += val;
}

int get(int r) {
    int res = 0;
    for(; r > 0; r -= r & -r) res += t[r];
    return res;
}

ll count_swaps(veci b) {
    n = sz(b) / 2;
    for(int i = 0; i < 2 * n; ++i) {
        upd(i + 1, 1);
        if(b[i] < 0) pos[-b[i]][0] = i + 1;
        else pos[b[i]][1] = i + 1;
    }
    ll ans = 0;
    vector <pii> seg;
    for(int i = 1; i <= n; ++i) {
        int l = pos[i][0], r = pos[i][1];
        if(l > r) ++ans, swap(l, r);
        seg.pb(mp(l, r));
    }
    sort(all(seg));
    for(auto p : seg) {
        int l = p.f, r = p.s;
        ans += get(r - 1) - get(l);
        upd(l, -1);
        upd(r, -1);
    }
    return ans;
}

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