Submission #677080

#TimeUsernameProblemLanguageResultExecution timeMemory
677080penguin133Arranging Shoes (IOI19_shoes)C++17
10 / 100
14 ms19092 KiB
#include <bits/stdc++.h>
#include "shoes.h"
using namespace std;

//#define int long long
#define pi pair<int, int>
#define pii pair<int, pi>
#define fi first
#define se second
#ifdef _WIN32
#define getchar_unlocked _getchar_nolock
#endif

set<int> L[200005], R[200005];
int vis[200005];

long long count_swaps(std::vector<int> s) {
	int n = (int)s.size();
	for(int i=0;i<n;i++){
		if(s[i] < 0)L[-s[i]].insert(i);
		else R[s[i]].insert(i);
	}
	
	long long ans = 0;
	
	for(int i=0;i<n;i++){
		if(vis[i])continue;
		if(s[i] < 0){
			L[-s[i]].erase(L[-s[i]].begin());
			ans += max(0, *R[-s[i]].begin() - i - 1);
			vis[*R[-s[i]].begin()] = 1;
			R[-s[i]].erase(R[-s[i]].begin());
		}
		else{
			ans++;
			R[s[i]].erase(R[s[i]].begin());
			ans += max(0, *L[s[i]].begin() - i - 1);
			vis[*L[s[i]].begin()] = 1;
			L[s[i]].erase(L[s[i]].begin());
		}
	}
	
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...