Submission #204344

#	Time	Username	Problem	Language	Result	Execution time	Memory
204344		ADJA	Arranging Shoes (IOI19_shoes)	C++14	45 / 100	87 ms	12152 KiB

shoes

#include "shoes.h"

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include <bitset>
#include <cmath>
#include <queue>
#include <stack>
#include <string>

#define sz(x) (int) x.size()
#define all(x) x.begin(), x.end()

using namespace std;

int n;
vector<int> s;

long long solveSameSize() {
  set<int> stA, stB;
  for (int i = 0; i < n; i++) {
    if (s[i] < 0) {
      stA.insert(i);
    } else {
      stB.insert(i);
    }
  }

  long long result = 0;
  for (int i = 0; i < n; i++) {
    if (i % 2 == 0) {
      int mn = *stA.begin();
      stA.erase(stA.begin());
      result += max(0, mn - i);
    } else {
      int mn = *stB.begin();
      stB.erase(stB.begin());
      result += max(0, mn - i);
    }
  }

  return result;
}

long long solveFour() {
  long long result = 0;

  for (int i = n / 2 - 1; i >= 0; i--) {
    result += i;
  }

  return result;
}

long long solveSmall() {
  return 0;
}

long long count_swaps(std::vector<int> S) {
  s = S;
  n = sz(s);

  int sz = abs(s[0]);
  bool sameSize = true;
  for (int i = 0; i < n; i++) {
    if (abs(s[i]) != sz) {
      sameSize = false;
    }
  }
  if (sameSize) {
    return solveSameSize();
  }

  bool ok = true;
  for (int i = 0; i < n / 2; i++) {
    if (s[i] > 0 || abs(s[i]) != abs(s[i + n / 2])) {
      ok = false;
    }
  }
  for (int i = n / 2; i < n; i++) {
    if (s[i] < 0) {
      ok = false;
    }
  }
  if (ok) {
    return solveFour();
  }

  return solveSmall();
}

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