# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
855817 | dong_liu | Arranging Shoes (IOI19_shoes) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define ar array
#define sz(v) int(v.size())
typedef long long ll;
typedef pair<int, int> pi;
struct FT {
vector<int> t;
FT(int n) : t(n,0) {}
void upd(int i, int x) {
while (i < sz(t)) {
t[i] += x;
i |= i+1;
}
}
int qry(int i) {
int x = 0;
while (i >= 0) {
x += t[i];
i &= i+1, i--;
}
return x;
}
int qry(int l, int r) {
return qry(r)-qry(l-1);
}
};
ll count_swap(vector<int> v) {
int n = sz(v) / 2;
vector<vector<int>> L(n), R(n);
for (int i = 0; i < n*2; i++)
if (v[i]<0) L[abs(v[i])-1].push_back(i);
else R[abs(v[i])-1].push_back(i);
vector<pi> f;
for (int a = 0; a < n; a++) {
assert(sz(L[a]) == sz(R[a]));
for (int j = 0; j < sz(L[a]); j++)
f.push_back({L[a][j],R[a][j]});
}
ll ans = 0;
for (auto &[l,r] : f) {
if (l>r) {
swap(l,r);
ans++;
}
}
sort(f.begin(), f.end());
FT ft(n*2);
for (auto [l,r] : f) {
ans += ft.qry(l,n*2-1)+ft.qry(r,n*2-1);
ft.upd(r,1);
}
return ans;
}
// int main() {
// cout << count_swap({2,1,-1,-2}) << '\n';
// cout << count_swap({-2,2,2,-2,-2,2}) << '\n';
// }