Submission #361855

#	Time	Username	Problem	Language	Result	Execution time	Memory
361855		Matteo_Verz	Arranging Shoes (IOI19_shoes)	C++17	10 / 100	1096 ms	364 KiB

shoes

#include "shoes.h"
#include <bits/stdc++.h>

using namespace std;
const int N = 1e5;

class Fenwick {
  private:
    vector <int> a;

    int lsb(int i) {
      return (i & (-i));
    }

  public:
    Fenwick(int n) {
      a.resize(1 + n);
    }

    void Update(int pos, int val) {
      for(; pos < a.size(); pos += lsb(pos))
        a[pos] += val;
    }

    int Query(int pos) {
      int ans(0);
      for(; pos > 0; pos -= lsb(pos))
        ans += a[pos];

      return ans;
    }
};

int rightpos[1 + N];
long long count_swaps(vector<int> s) {
  map <int, int> mp;
  s.insert(s.begin(), 0);
  int n = s.size() - 1;
  Fenwick aib(n);

  for(int i = 1; i <= n; i++) {
    if(mp.find(-s[i]) != mp.end()) {
      rightpos[mp[-s[i]]] = i;
      rightpos[i] = i;
      mp.erase(-s[i]);
    } else mp.insert(make_pair(s[i], i));
    aib.Update(i, 1);
  }

  long long ans(0);
  for(int i = 1; i <= n; i++) {
    int pereche = rightpos[i];
    if(pereche != i) {
      aib.Update(i, -1); aib.Update(pereche, -1);

      if(s[i] < 0) ans += 1LL * aib.Query(pereche);
      else ans += 1LL * (aib.Query(pereche) + 1);
    }
  }

  return ans;
}

Compilation message (stderr)

shoes.cpp: In member function 'void Fenwick::Update(int, int)':
shoes.cpp:21:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |       for(; pos < a.size(); pos += lsb(pos))
      |             ~~~~^~~~~~~~~~

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