Submission #811103

#	Time	Username	Problem	Language	Result	Execution time	Memory
811103		OAleksa	Arranging Shoes (IOI19_shoes)	C++14	10 / 100	10 ms	19892 KiB

shoes

#include <bits/stdc++.h>
#include "shoes.h"
 
#define f first
#define s second
using namespace std; 
 
const int maxn = 2e5 + 69;
set<int> l[maxn], r[maxn];
vector<int> f(maxn);

void add(int v, int x) {
	for(int i = v;i < maxn;i += (i & -i))
		f[i] += x;
}

int qry(int v) {
	int res = 0;
	for(int i = v;i >= 1;i -= (i & -i)) 
		res += f[i];
	return res;
}

long long count_swaps(vector<int> s) {
	int n = s.size();
	vector<int> a(n + 1);
	for(int i = 0;i < n;i++) {
		a[i + 1] = s[i];
	}
	vector<int> vis(n + 1);
	vector<pair<int, int>> par;
	for(int i = 1;i <= n;i++) {
		if(a[i] < 0)
			l[abs(a[i])].insert(i);
		else
			r[a[i]].insert(i);
	}
	for(int i = 1;i <= n;i++) {
		if(a[i] < 0 && !vis[i]) {
			auto u = *r[abs(a[i])].upper_bound(i);
			par.push_back({i, u});
			vis[i] = vis[u] = 1;
		}
		else if(a[i] > 0 && !vis[i]) {
			auto u = *l[a[i]].upper_bound(i);
			par.push_back({i, u});
			vis[i] = vis[u] = 1;
		}
	}
	long long ans = 0;
	for(auto x : par) {
		int levi = x.f + qry(x.f);
		int desni = x.s + qry(x.s);
		if(a[x.f] < 0) 
			ans += (desni - levi - 1);
		else 
			ans += (desni - levi);
		add(x.f + 1, 1);
		add(x.s, -1);
	}
	return ans;
}
 
 
 
// int main()
// {
	  // ios_base::sync_with_stdio(false);
	  // cin.tie(0);
	  // cout.tie(0);
	  // int tt = 1;
		// //cin >> tt;
	  // while(tt--) {
			// int n;
			// cin >> n;
			// vector<int> a(2 * n);
			// for(int i = 0;i < 2 * n;i++)
				// cin >> a[i];
			// cout << count_swaps(a) << endl;
		// }
    // return 0;
// }

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