Submission #143797

#	Time	Username	Problem	Language	Result	Execution time	Memory
143797		Nightlight	Arranging Shoes (IOI19_shoes)	C++14	100 / 100	223 ms	137500 KiB

shoes

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

queue<int> pass[200005];
int tree[200005], MAXS;
long long ans;

void update(int u, int idx){
  idx++;
  for(int i = idx; i <= MAXS; i += i & -i){
    tree[i] += u;
  }
}

int getans(int idx){
  idx++;
  int res = 0;
  for(int i = idx; i > 0; i -= i & -i)
    res += tree[i];
  return res;
}

long long count_swaps(vector<int> s) {
  MAXS = s.size();
	for(int R = 0; R < MAXS; R++){
    if(!pass[100000-s[R]].empty()){
      int L = pass[100000-s[R]].front();
//      cout << R << " " << L << "\n";
      pass[100000-s[R]].pop();
      if(s[R] > 0){
        ans += R - L - 1; 
      }else ans += R - L;

      ans -= getans(R) - getans(L);
      update(-1, L);
    }else{
      pass[100000 + s[R]].push(R);
      update(1, R);
    }
  }
  return ans;
}

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