| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 527320 | Olympia | Arranging Shoes (IOI19_shoes) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <cmath>
#include <cassert>
#include <iostream>
#include <set>
#include <climits>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <type_traits>
#include <string>
#include <queue>
#include <map>
//#include "shoes.h"
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("Ofast")
#pragma GCC optimization ("unroll-loops")
using namespace std;
struct segmentTree {
vector<int64_t> v;
vector<int64_t> val;
int64_t ID = 0;
int64_t op(int64_t a, int64_t b) {
return a + b;
}
int64_t query(int dum, int tl, int tr, int &l, int &r) {
if (tr < l || tl > r) {
return ID;
}
if (tl >= l && tr <= r) {
return val[dum];
}
int64_t mid = (tl + tr) >> 1;
dum = dum << 1;
return op(query(dum, tl, mid, l, r), query(dum + 1, mid + 1, tr, l, r));
}
int64_t query(int l, int r) {
return query(1, 0, (int) v.size() - 1, l, r);
}
void update(int x, int64_t y) {
int cur = (int) v.size() + x;
int curX = x;
int curY = x;
while (true) {
val[cur] = op(op(query(curX, x - 1), query(x + 1, curY)), y);
if (cur == 0) {
break;
}
if (cur % 2 == 0) {
curY = 2 * curY - curX + 1;
} else {
curX = 2 * curX - curY - 1;
}
cur /= 2;
}
}
void resz(int n) {
v.resize((1 << (int) ceil(log2(n))));
val.resize(v.size() * 2);
}
};
int64_t get_inversions(vector<int> v) {
segmentTree st;
st.resz(v.size() + 3);
int64_t ans = 0;
for (int i = 0; i < v.size(); i++) {
ans += st.query(v[i] + 1, v.size() + 2);
st.update(v[i], st.query(v[i], v[i]) + 1);
}
return ans;
}
long long count_swaps(vector<int> v) {
map<int, set<int>> myMap;
for (int i = 0; i < v.size(); i++) {
myMap[v[i]].insert(i);
}
vector<int> vec(v.size());
int cntr = 0;
for (int i = 0; i < v.size(); i++) {
if (myMap[-v[i]].empty()) continue;
if (myMap[v[i]].count(i) == false) continue;
int x = *myMap[-v[i]].begin();
myMap[-v[i]].erase(x);
myMap[v[i]].erase(i);
vec[i] = cntr++;
vec[x] = cntr++;
if (v[i] > 0) {
swap(vec[i], vec[x]);
}
}
for (int i: vec) {
cout << i << ' ';
}
cout << '\n';
return get_inversions(vec);
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout << count_swaps({-1,1,-1,-1,-1,1,1,1});
}