# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
235945 | triple_fault | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "shoes.h"
#include <set>
#include <cstdio>
#include <algorithm>
#define ll long long
#define mid ((tl + tr) / 2)
#define lft (v * 2)
#define rht ((v * 2) + 1)
#define MAX 200000
using namespace std;
ll n;
vector<ll> shoes(MAX);
struct segtree {
segtree* left = NULL; segtree* right = NULL;
multiset<ll> here;
ll m_tl, m_tr;
bool inactive = true;
ll sum_here = 0;
void build(ll tl, ll tr) {
m_tl = tl, m_tr = tr;
if (tl == tr) {
here.insert(shoes[tl]);
inactive = false;
return;
}
left = new segtree();
right = new segtree();
left->build(tl, mid);
right->build(mid + 1, tr);
merge(left->here.begin(), left->here.end(), right->here.begin(), right->here.end(),
inserter(here, here.begin()));
}
ll query(ll needle) {
here.erase(here.find(needle));
++sum_here;
if (m_tl == m_tr) {
inactive = true;
return m_tl;
}
if (left->here.find(needle) != left->here.end())
return left->query(needle);
else if (right->here.find(needle) != right->here.end())
return right->query(needle);
else puts("fuck this");
return MAX * MAX;
}
ll sum(ll tl, ll tr) {
if (tl > tr) return 0;
if (tl == m_tl && tr == m_tr) return sum_here;
return left->sum(tl, min(tr, (m_tl + m_tr) / 2)) +
right->sum(max(tl, ((m_tl + m_tr) / 2) + 1), tr);
}
} segtree;
long long count_swaps(std::vector<int> s) {
ll n = s.size();
for (ll i = 0; i < n; ++i) shoes[i] = (ll)s[i];
segtree.build(0, n - 1);
ll inactive_a[n];
memset(inactive_a, 0, sizeof inactive_a);
ll ans = 0;
for (ll i = 0; i < n; ++i) {
if (inactive_a[i]) continue;
segtree.query(shoes[i]);
ll nxt = -shoes[i];
ll q = segtree.query(nxt);
ans += (q - i - 1 - segtree.sum(i + 1, q - 1));
if (nxt < 0) ++ans;
inactive_a[i] = 1; inactive_a[q] = 1;
}
return ans;
}