답안 #766439

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
766439 2023-06-25T16:14:12 Z Chal1shkan Tree Rotations (POI11_rot) C++14
100 / 100
159 ms 16096 KB
//Bismillahir-Rahmanir-Rahim
 
# include <bits/stdc++.h>
 
# define pb push_back
# define ff first
# define ss second
# define nl "\n"
# define sz(x) ((int)(x).size())
# define all(x) (x).begin(), (x).end()
# define deb(x) cerr << #x  << " = " << x << endl; 
# define pll pair <ll, ll>
# define pii pair <int, int>
 
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
 
const ll maxn = 2e5 + 2;
const ll inf = 2e18 + 0;
const ll mod = 1e9 + 7;
const ll dx[] = {-1, 1, 0, 0};
const ll dy[] = {0, 0, -1, 1};
 
using namespace std;
 
ll n, nv, ans;
ll L[maxn * 2];
ll R[maxn * 2];
ll sz[maxn * 2];
ll cnt[maxn * 2];
ll t[maxn];
 
void upd (ll id, ll x)
{
    for (; id < maxn; id += (id & (-id)))
    {
        t[id] += x;
    }
}
 
ll get (ll id)
{
    ll res = 0;
    for (; id >= 1; id -= (id & (-id)))
    {
        res += t[id];
    }
    return res;
}
 
ll rec ()
{
    ll x; cin >> x;
    if (!x)
    {
        ll v = ++nv;
        L[v] = rec ();
        R[v] = rec ();
        sz[v] = sz[L[v]] + sz[R[v]] + 1;
        cnt[v] = cnt[L[v]] + cnt[R[v]];
        return v;
    }
    sz[x] = 1, cnt[x] = 1;
    return x;
}
 
void add (ll v)
{
    if (v <= n) upd(v, 1);
    else add(L[v]), add(R[v]);
}
 
void clr (ll v)
{
    if (v <= n) upd(v, -1);
    else clr(L[v]), clr(R[v]);
}
 
ll calc (ll v)
{
    if (v <= n) return get(v);
    return calc(L[v]) + calc(R[v]);
}
 
void dfs (ll v = n + 1, bool keep = 1)
{
    if (v <= n)
    {
        add(v);
    }
    else
    {
        if (sz[L[v]] < sz[R[v]])
        {
            swap(L[v], R[v]);
        }
        dfs(R[v], 0);
        dfs(L[v], 1);
        ll inv = calc(R[v]);
      //  cout << v << ' ' << inv << ' ' << cnt[L[v]] << ' ' << cnt[R[v]] << nl;
        ans += min(1LL * inv, cnt[L[v]] * 1LL * cnt[R[v]] - inv);
        add(R[v]);
    }
    if (!keep)
    {
        clr(v);
    }
}
 
void ma1n (/* SABR */)
{
    cin >> n;
    nv = n;
    rec ();
    dfs ();
    cout << ans;
}
 
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
//  freopen("file.in", "r", stdin);
//  freopen("file.out", "w", stdout);
    int ttt = 1;
//  cin >> ttt;
    for (int test = 1; test <= ttt; ++test)
    {
//      cout << "Case " << test << ":" << '\n';
        ma1n();
    }
    return 0;
}
 
// 998batrr | BbIWEJI 3A TObOU!!!
// tourist  | BbIWEJI 3A TObOU!!!
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 0 ms 340 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 0 ms 340 KB Output is correct
4 Correct 0 ms 340 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 724 KB Output is correct
2 Correct 2 ms 576 KB Output is correct
3 Correct 1 ms 724 KB Output is correct
4 Correct 1 ms 724 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 1492 KB Output is correct
2 Correct 6 ms 980 KB Output is correct
3 Correct 18 ms 2004 KB Output is correct
4 Correct 6 ms 1364 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 29 ms 3152 KB Output is correct
2 Correct 17 ms 3932 KB Output is correct
3 Correct 22 ms 4860 KB Output is correct
4 Correct 18 ms 4924 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 19 ms 8516 KB Output is correct
2 Correct 23 ms 6924 KB Output is correct
3 Correct 33 ms 5896 KB Output is correct
4 Correct 24 ms 5460 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 34 ms 6996 KB Output is correct
2 Correct 34 ms 7852 KB Output is correct
3 Correct 33 ms 9860 KB Output is correct
4 Correct 36 ms 9736 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 110 ms 11696 KB Output is correct
2 Correct 78 ms 11000 KB Output is correct
3 Correct 66 ms 11028 KB Output is correct
4 Correct 71 ms 10700 KB Output is correct
5 Correct 126 ms 9792 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 59 ms 11504 KB Output is correct
2 Correct 84 ms 15052 KB Output is correct
3 Correct 77 ms 13756 KB Output is correct
4 Correct 57 ms 15612 KB Output is correct
5 Correct 148 ms 11096 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 75 ms 11768 KB Output is correct
2 Correct 66 ms 12524 KB Output is correct
3 Correct 110 ms 11444 KB Output is correct
4 Correct 96 ms 11900 KB Output is correct
5 Correct 56 ms 16096 KB Output is correct
6 Correct 159 ms 11404 KB Output is correct