제출 #795879

#제출 시각아이디문제언어결과실행 시간메모리
795879vjudge1Arranging Shoes (IOI19_shoes)C++17
50 / 100
14 ms1876 KiB
#include "shoes.h"
#include "bits/stdc++.h"

using namespace std;

const int tam = 200010;
int bit[tam];
void add(int pos, int val)
{
	while(pos < tam)
		bit[pos] += val, pos += pos & -pos;
}
int query(int pos)
{
	int res = 0;
	while(pos)
		res += bit[pos], pos -= pos & -pos;
	return res;
}

long long count_swaps(std::vector<int> s) {
	int n = (int)s.size() / 2;
	if(n > 1000) return 0;
	// cout<<n<<'\n';
	vector<queue<int>> colasp(n + 1), colasn(n + 1);
	vector<pair<int, int>> pares;
	// cout<<s.size()<<'\n';
	for(int i = 0; i < 2 * n; i++)
	{
		// cout<<i<<'\n';
		add(i + 1, 1);
		if(s[i] > 0)
		{
			if(colasn[s[i]].empty())
				colasp[s[i]].push(i);
			else
				pares.push_back({colasn[s[i]].front(), i}), colasn[s[i]].pop();
		}
		else
		{
			if(colasp[-s[i]].empty())
				colasn[-s[i]].push(i);
			else
				pares.push_back({colasp[-s[i]].front(), i}), colasp[-s[i]].pop();
		}
	}
	long long res = 0;
	sort(pares.begin(), pares.end());
	// cout<<pares.size()<<'\n';
	for(int i = 0; i < n; i++)
	{
		// cout<<i<<'\n';
		int a = pares[i].first, b = pares[i].second;
		res += query(b) - query(a + 1);
		if(s[a] > 0)
			res++;
		add(b + 1, -1);
		add(a + 1, 1);
		
	}
	return res;
}
#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...