Submission #984221

#	Time	Username	Problem	Language	Result	Execution time	Memory
984221		SzymonKrzywda	Arranging Shoes (IOI19_shoes)	C++17	45 / 100	62 ms	12232 KiB

shoes

//#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;

const int base = 2<<17;
int tree[base*2+7];

void edit(int n){
    n+=base;
    while (n>0){
        tree[n] += 1;
        n/=2;
    }
}

int query(int a,int b){
    a+=base-1;
    b+=base+1;
    int wynik=0;
    while (a/2 != b/2){
        if (a%2==0) wynik += tree[a+1];
        if (b%2==1) wynik += tree[b-1];
        a/=2;
        b/=2;
    }
    return wynik;
}


long long count_swaps(vector<int>S){
    long long wynik = 0,idx_2=0,val=0,idx_3=0,size_=0,i=0;
    pair< vector<int>,int> idx[100007];
    
    priority_queue<pair<int,int>> kolejka;
    //pos, item
    for (int i=0; i<S.size(); ++i){
        if (S[i]>0) idx[S[i]].first.push_back(i);
        else kolejka.push({-i,S[i]});
    }
    size_ = S.size();
    
    while (!kolejka.empty()){
        idx_2 = -kolejka.top().first;
        val = kolejka.top().second;
        kolejka.pop();
        //cout << idx_2 << " " << val << endl;
        
        wynik += idx_2-i+query(idx_2+1,size_);
        edit(idx_2);
        //cout << wynik << endl;
        
        
        idx_3 = idx[-val].first[idx[-val].second];
        wynik += idx_3-(i+1)+query(idx_3+1,size_);
        edit(idx_3);
        //cout << wynik << endl << endl;
        idx[-val].second += 1;
        i+=2;
    }
    
    for (int j=0; j<base*2+7; j++){
        tree[j] = 0;
    }
    return wynik;
}

Compilation message (stderr)

shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:36:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     for (int i=0; i<S.size(); ++i){
      |                   ~^~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 20
• Subtask 3: 20 / 20
• Subtask 4: 15 / 15
• Subtask 5: 0 / 20
• Subtask 6: 0 / 15