# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
201414 | khulegub | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "shoes.h"
#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define xx first
#define yy second
#define ll(x) (x*2) + 1
#define rr(x) (x*2) + 2
using namespace std;
typedef long long i64;
vector<int> ft;
int query(int x){
int sm = 0;
while (x){
sm += ft[x];
x -= x & (-x);
}
return sm;
}
int query(int l, int r){
int bb = 0;
if (l != 0) bb = query(b - 1);
return query(r) - bb;
}
int update(int x, int v){
while (x){
ft[x] += v;
x += x & (-x);
}
}
i64 count_swaps(std::vector<int> s) {
i64 ans = 0LL;
int n = s.size();
ft.resize(n);
vector<int> maap_l[100005], maap_r[100005];
for (int i = n - 1; i >= 0; i--){
if (s[i] < 0) maap_l[s[i] * -1].pb(i);
else maap_r[s[i]].pb(i);
}
vector<bool> taken(n, 0);
for (int i = 0; i < n; i++){
int tmp;
if (!taken[i]){
if (s[i] < 0){
maap_r[s[i]].pop_back();
tmp = maap_r[s[i]].back();
}
else{
maap_l[s[i]].pop_back();
tmp = maap_l[s[i]].back();
}
taken[tmp] = 1;
if (s[i] < 0){
ans += tmp - i - 1;
}
else{
ans += tmp - i;
}
ans -= query(i, tmp);
update(tmp, 1);
}
}
return ans;
}