Submission #44116

#	Time	Username	Problem	Language	Result	Execution time	Memory
44116		imaxblue	Untitled (POI11_rot)	C++17	100 / 100	372 ms	40368 KiB

Main

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mp make_pair
#define pb push_back
#define x first
#define y second
#define pii pair<int, int>
#define p3i pair<pii, int>
#define pll pair<ll, ll>
#define p3l pair<pll, ll>
#define lseg L, (L+R)/2, N*2+1
#define rseg (L+R)/2+1, R, N*2+2
#define ub upper_bound
#define lb lower_bound
#define pq priority_queue
#define MN 1000000007
#define fox(k, x) for (int k=0; k<x; ++k)
#define fox1(k, x) for (int k=1; k<=x; ++k)
#define foxr(k, x) for (int k=x-1; k>=0; --k)
#define fox1r(k, x) for (int k=x; k>0; --k)
#define ms multiset
#define flood(x) memset(x, 0x3f3f3f3f, sizeof x)
#define drain(x) memset(x, 0, sizeof x)
#define rng() ((rand() << 14)+rand())
#define scan(X) do{while((X=getchar())<'0'); for(X-='0'; '0'<=(_=getchar()); X=(X<<3)+(X<<1)+_-'0');}while(0)
char _;
#define pi 3.14159265358979323846

int bit[200005], a[200005], sub[400005], n, t, p, lft[400005], rit[400005];
void add(int P, int V){
    for(; P<=200000; P+=P&-P) bit[P]+=V;
}
int sum(int P, int C=0){
    for(; P>0; P-=P&-P) C+=bit[P];
    return C;
}
int read(){
    int t;
    cin >> t;
    if (t!=0){
        sub[t]=1;
        return t;
    }
    t=(p++);
    lft[t]=read();
    rit[t]=read();
    sub[t]=1+sub[lft[t]]+sub[rit[t]];
}
void clr(int N){
    if (N<=n){
        if (a[N]){
            a[N]=0;
            add(N, -1);
        }
        return;
    }
    clr(lft[N]);
    clr(rit[N]);
}
ll ans, c, c2;
void query(int N){
    if (N<=n){
        c+=sum(N);
        c2+=sum(200000);
        return;
    }
    query(lft[N]);
    query(rit[N]);
}
void on(int N){
    if (N<=n){
        a[N]=1;
        add(N, 1);
        return;
    }
    on(lft[N]);
    on(rit[N]);
}
void dfs(int N){
    if (N<=n){
        a[N]=1;
        add(N, 1);
        return;
    }
    int a, b;
    if (sub[lft[N]]>sub[rit[N]]){
        a=lft[N]; b=rit[N];
    } else {
        a=rit[N]; b=lft[N];
    }
    dfs(b); clr(b);
    dfs(a);
    c=c2=0;
    query(b); on(b);
    ans+=min(c, c2-c);
}
int main(){
    cin >> n;
    p=n+1;
    read();
    dfs(n+1);
    cout << ans;
    return 0;
}

Compilation message (stderr)

rot.cpp: In function 'int read()':
rot.cpp:49:1: warning: control reaches end of non-void function [-Wreturn-type]
 }
 ^

• Subtask 1: 9 / 9
• Subtask 2: 9 / 9
• Subtask 3: 9 / 9
• Subtask 4: 9 / 9
• Subtask 5: 9 / 9
• Subtask 6: 9 / 9
• Subtask 7: 9 / 9
• Subtask 8: 9 / 9
• Subtask 9: 9 / 9
• Subtask 10: 9 / 9
• Subtask 11: 10 / 10