| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 317989 | Drew_ | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
//#include "shoes.h"
#include <cstdio>
#include <cassert>
#include <vector>
#include <queue>
#include <iostream>
using namespace std;
int const N = 1e5 + 7;
int bit[2*N] = {};
void add(int x, int val)
{
x++;
while (x < 2*N)
bit[x] += val, x += (x & -x);
}
int sum(int x)
{
x++;
int res = 0;
while (x)
res += bit[x], x -= (x & -x);
return res;
}
queue<int> q[N];
bool type[N]; //false: left, true: right
long long count_swaps(std::vector<int> s)
{
long long res = 0;
for (int i = 0; i < s.size(); ++i)
{
if (q[abs(s[i])].empty() || !type[abs(s[i])] && s[i] < 0 || type[abs(s[i])] && s[i] > 0)
{
q[abs(s[i])].push(i);
if (s[i] < 0) type[abs(s[i])] = false;
else type[abs(s[i])] = true;
}
else
{
int target = q[abs(s[i])].front();
q[abs(s[i])].pop();
//cout << i << " -> " << target << ", shifted: " << sum(target) << ", cost: " << i - target - sum(target) << '\n';
res += i - target - sum(target);
if (s[i] > 0) res--;
add(target, 1);
add(i, -1);
}
}
return res;
}
int main() {
int n;
assert(1 == scanf("%d", &n));
vector<int> S(2 * n);
for (int i = 0; i < 2 * n; i++)
assert(1 == scanf("%d", &S[i]));
fclose(stdin);
long long result = count_swaps(S);
printf("%lld\n", result);
fclose(stdout);
return 0;
}