Submission #1364823

#TimeUsernameProblemLanguageResultExecution timeMemory
1364823nataliaaArranging Shoes (IOI19_shoes)C++20
10 / 100
61 ms137436 KiB
#include<bits/stdc++.h>
// #include "shoes.h"
using namespace std;
const int MAXN = 100005;

long long ok[MAXN] = {};
void upd(int idx, int val){
    while(idx<=MAXN){
        ok[idx]+=val;
        idx+=idx&(-idx);
    }
}

long long get(int idx){
    long long s= 0;
    while(idx>0){
        s+=ok[idx];
        idx-=idx&(-idx);
    }
    return s;
}
long long count_swaps(vector<int> v) {
    long long ans = 0;
    queue<int> p[MAXN][2]={};
    for(int i = 0; i < v.size(); i++){
        if(v[i]<0) p[-v[i]][0].push(i);
        else{ p[v[i]][1].push(i);}
    }
    int es[2*MAXN] = {};
    for(long long i = 0; i < v.size(); i++){
        if(es[i]) continue;
        if(v[i]<0){
            long long j = p[-v[i]][1].front();
            long long k1 =i+ get(i+1), k2 =j+ get(j+1);
            ans+=k2-k1-1;
            p[-v[i]][1].pop();
            p[-v[i]][0].pop();
            es[i] = 1;
            es[j] = 1;
            upd(k1+1, 1);
            upd(k2+2, -1);
            continue;
            
        }
        ans++;
        long long j = p[v[i]][0].front();
        long long k1 =i+ get(i+1), k2 =j+ get(j+1);
        ans+=k2-k1-1;
        es[i] = 1;
        es[j] = 1;
        p[v[i]][0].pop();
        p[v[i]][1].pop();
        upd(k1+1, 1);
        upd(k2+2, -1);
    }
    return ans;
}
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...