| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1335237 | zhehan | Arranging Shoes (IOI19_shoes) | C++20 | 0 ms | 0 KiB |
#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;
#define int long long
vector<int> ft;
void upd(int p, int v) {
while (p < ft.size()) {
ft[p] += v;
p += p & (-p);
}
}
int query(int p) {
int sum = 0;
while (p > 0) {
sum += ft[p];
p -= p & (-p);
}
return sum;
}
int invquery(int p) { return query(ft.size() - 1) - query(p); }
long long count_swaps(std::vector<signed> s) {
int n = s.size() / 2;
map<int, queue<int>> ind;
vector<int> inv(2 * n, 0);
int c = 1;
int tinv = 0;
for (int i = 0; i < 2 * n; ++i) {
if (ind[-s[i]].empty()) {
inv[i] = 2 * c - 1;
ind[s[i]].push(i);
++c;
if (s[i] > 0)
tinv++;
} else {
inv[i] = inv[ind[-s[i]].front()] + 1;
ind[-s[i]].pop();
}
}
ft = vector<int>(2 * n + 10, 0);
for (auto e : inv) {
tinv += invquery(e);
upd(e, 1);
}
return tinv;
}
signed main() {
int n;
cin >> n;
vector<signed> s(2 * n, 0);
for (int i = 0; i < 2 * n; ++i) {
cin >> s[i];
}
cout << count_swaps(s);
return 0;
}
