Submission #644402

# Submission time Handle Problem Language Result Execution time Memory
644402 2022-09-24T15:36:27 Z a_aguilo Arranging Shoes (IOI19_shoes) C++14
10 / 100
1 ms 340 KB
#include "shoes.h"
#include<bits/stdc++.h>

using namespace std;

int N;

void change(int index, int delta, int BIT[]){
    for(; index <N; index += (index&(-index))){
        BIT[index]+=delta;
    }
}

int query(int index, int BIT[]){
    int ans = 0;
    for(; index > 0; index -= (index&(-index))) ans+= BIT[index];
    return ans;
}

long long count_swaps(vector<int> s) {
    map<int, vector<int>> M;
    N = s.size()+1;
    int BIT[N];
    memset(BIT, 0, sizeof(BIT));
    for(int i = 1; i < N; ++i) change(i, 1, BIT);
    long long ans = 0;
    for(int i = 0; i < s.size(); ++i){
        //cout << i << " " << ans << endl;
        if(M.find(-s[i]) == M.end()){
            if(M.find(s[i]) == M.end()){
                M[s[i]] = {i};
            }
            else if(M[s[i]].empty()){
                M[s[i]] = {i};
            }
            else{
                M[s[i]].push_back(i);
            }
        }
        else if(M[-s[i]].empty()){
            if(M.find(s[i]) == M.end()){
                M[s[i]] = {i};
            }
            else if(M[s[i]].empty()){
                M[s[i]] = {i};
            }
            else{
                M[s[i]].push_back(i);
            }
        }
        else{
            if(s[i] < 0) ans++;
            //cout << "a ";
            int prev = M[-s[i]].back();
            //cout <<"b ";
            M[-s[i]].pop_back();
            //cout <<"c ";
            ans += (long long)query(i, BIT)-(long long)query(prev+1, BIT);
            //cout <<"d ";
            change(i+1, -1, BIT);
            //cout << "e ";
            change(prev+1, 1, BIT);
        }

    }
	return ans;
}

Compilation message

shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:27:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |     for(int i = 0; i < s.size(); ++i){
      |                    ~~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 300 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 300 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 1 ms 300 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 1 ms 212 KB Output is correct
14 Incorrect 0 ms 212 KB Output isn't correct
15 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 300 KB Output is correct
3 Incorrect 1 ms 212 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 1 ms 212 KB Output is correct
4 Incorrect 0 ms 212 KB Output isn't correct
5 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 300 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 1 ms 300 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 1 ms 212 KB Output is correct
14 Incorrect 0 ms 212 KB Output isn't correct
15 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 300 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 1 ms 300 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 1 ms 212 KB Output is correct
14 Incorrect 0 ms 212 KB Output isn't correct
15 Halted 0 ms 0 KB -