| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 645360 | _petar_b | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "shoes.h"
#include <bits/stdc++.h>
#define MAXN 200010
#define DELTA 100001
#define pb push_back
#define ll long long
#define fi first
#define se second
#define mp make_pair
//using namespace std;
int n, t[4*MAXN], par[MAXN], tleaf[MAXN];
queue<int> posOf[MAXN];
void build(int v, int tl, int tr)
{
if (tl == tr)
t[v] = 1;
else
{
int tm = (tl + tr) / 2;
build(v*2, tl, tm);
build(v*2+1, tm+1, tr);
t[v] = t[v*2] + t[v*2+1];
}
}
int sum(int v, int tl, int tr, int l, int r)
{
if (l > r)
return 0;
if (l == tl && r == tr)
return t[v];
int tm = (tl + tr) / 2;
return sum(v*2, tl, tm, l, min(r, tm)) + sum(v*2+1, tm+1, tr, max(l, tm+1), r);
}
void update(int v, int tl, int tr, int pos)
{
if (tl == tr)
t[v] = 0;
else
{
int tm = (tl + tr) / 2;
if (pos <= tm)
update(v*2, tl, tm, pos);
else
update(v*2+1, tm+1, tr, pos);
t[v] = t[v*2] + t[v*2+1];
}
}
long long count_swaps(std::vector<int> s)
{
n = s.size();
for (int i = n-1; i >= 0; i--)
{
tleaf[i] = 1;
int other = DELTA - s[i];
if (!posOf[other].empty())
{
par[i] = posOf[other].front();
par[par[i]] = i;
tleaf[par[i]] = 0;
posOf[other].pop();
}
else
{
par[i] = -1;
posOf[s[i]+DELTA].push(i);
}
}
build(1, 0, n-1);
int res = 0;
for (int i = 0; i < n; i++)
{
if (tleaf[i] == 1)
{
res += sum(1, 0, n-1, i+1, par[i]-1);
//tleaf[par[i]] = 0;
update(1, 0, n-1, par[i]);
if (s[i] > 0) res++;
}
}
return res;
}