# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1036887 | DeathIsAwe | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll segment[262144];
void update(ll a, ll b) {
a += 131072; b += 131072;
if (b >= 131072) {
return 1/0;
}
while (a <= b) {
if (a % 2 == 1) {segment[a++]++;}
if (b % 2 == 0) {segment[b--]++;}
a /= 2; b /= 2;
}
}
ll give(ll a) {
ll total = a;
if (a >= 131072) {
return 1/0;
}
a += 131072;
while (a > 0) {
total += segment[a];
a /= 2;
}
return total;
}
long long count_swaps(vector<int> s) {
for (int i=0;i<262144;i++) {
segment[i] = 0;
}
ll n = s.size(), shift = *max_element(s.begin(), s.end()), ans = 0;
vector<deque<ll>> positions(2 * shift + 1);
for (int i=0;i<n;i++) {
if (shift - s[i] < 0) {
return 1/0;
}
if (shift + s[i] >= positions.size()) {
return 1/0;
}
if (positions[shift - s[i]].empty()) {
positions[shift + s[i]].push_back(i);
} else {
ans += i - give(positions[shift - s[i]][0]);
update(positions[shift - s[i]][0], i);
if (s[i] > 0) {
ans--;
}
positions[shift - s[i]].pop_front();
}
}
return ans;
}