# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
311283 | saarang123 | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/stdc++.h"
using namespace std;
const int N = 2e5 + 2;
ll tree[N], n;
queue<int> l[N], r[N];
void upd(int x, int val) {
for(; x <= n; x += (x & -x))
tree[x] += val;
}
int sum(int x) {
if(x <= 0) return 0;
int ans = 0;
for(; x > 0; x -= (x & -x))
ans += tree[x];
return ans;
}
int qry(int l, int r) { return sum(r) - sum(l - 1); };
long long count_swaps(vector<int> a) {
n = a.size(); int i, j;
for(i = 0; i < n; i++) {
if(a[i] > 0) l[a[i]].push(i + 1);
else r[-a[i]].push(i + 1);
}
ll ans = 0;
for(i = 0; i < n; i++) {
if(a[i] > 0) {
if(l[a[i]].front() != i + 1) continue;
l[a[i]].pop();
int id = r[a[i]].front(); r[a[i]].pop();
ans += (id - i - 1 - qry(i + 1, id));
//cout << i << " " << a[i] << " " << id - i - qry(i + 1, id) << endl;
upd(id, 1);
}
else {
if(r[-a[i]].front() != i + 1) continue;
r[-a[i]].pop();
int id = l[-a[i]].front(); l[-a[i]].pop();
ans += (id - i - 2 - qry(i + 1, id));
//cout << i << " " << a[i] << " " << id - i - 1 - qry(i + 1, id) << endl;
upd(id, 1);
}
}
return ans;
}
/*
signed main(){
ios_base::sync_with_stdio(false);
cin.tie(NULL);cout.tie(0);
int i,j;
cin >> n;
vector<int> a(2 * n);
for(auto &x : a) cin >> x;
cout << count_swaps(a) << endl;
return 0;
}*/