Submission #971146

#	Time	Username	Problem	Language	Result	Execution time	Memory
971146		yellowtoad	Arranging Shoes (IOI19_shoes)	C++17	45 / 100	117 ms	143476 KiB

shoes

#include "shoes.h"
#include <iostream>
#include <queue>
#define f first
#define s second
#define int long long
using namespace std;

int p[200010], cnt, nxt, tmp[200010];
queue<pair<int,int>> l;
queue<int> r[200010];

void merge(int l, int r) {
	if (l >= r) return;
	int mid = (l+r)/2;
	merge(l,mid);
	merge(mid+1,r);
	int x = l, y = mid+1;
	for (int i = l; i <= r; i++) {
		if ((y > r) || ((x <= mid) && (p[x] < p[y]))) tmp[i] = p[x++];
		else {
			cnt += mid-x+1;
			tmp[i] = p[y++];
		}
	}
	for (int i = l; i <= r; i++) p[i] = tmp[i];
}

long long count_swaps(std::vector<signed> a) {
	for (int i = 0; i < a.size(); i++) {
		if (a[i] < 0) l.push({i,-a[i]});
		else r[a[i]].push(i);
	}
	for (int i = 0; i < a.size(); i++) {
		if (i%2 == 0) {
			p[l.front().f] = i;
			nxt = l.front().s;
			l.pop();
		} else {
			p[r[nxt].front()] = i;
			r[nxt].pop();
		}
	}
	merge(0,a.size()-1);
	return cnt;
}

Compilation message (stderr)

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