oj.uz

답안 #1116309

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1116309 2024-11-21T13:23:15 Z FubuGold Tree Rotations (POI11_rot) C++14
0 / 100
 274 ms 65536 KB 
#include <bits/stdc++.h>

using namespace std;

const int MAXN = 500000;

struct BIT {
    vector<int> tree;
    int n;
    void init(int _n) {
        n = _n;
        tree.assign(n+1,0);
    }
    void update(int pos,int val) {
        assert(pos > 0);
        for (;pos <= n;pos += pos & -pos) tree[pos] += val;
    }
    int query(int pos) {
        int sum = 0;
        for (;pos > 0;pos -= pos & -pos) sum += tree[pos];
        return sum;
    }
    int query(int l,int r) {
        if (l > r) return 0;
        return query(r) - query(l-1);
    }
} tree;

vector<int> adj[MAXN+1];
stack<pair<int,int>> st;
vector<int> ch[MAXN+1];
int val[MAXN+1];
int sz[MAXN+1];
int n;
int cnt;
long long ans = 0;

int cal_sz(int u) {
    if (adj[u].size() == 0) return sz[u] = 1;
    sz[u] += cal_sz(adj[u][0]);
    sz[u] += cal_sz(adj[u][1]);
    return sz[u];
}

void dfs(int u,int keep) {
    if (adj[u].size() == 0) {
        ch[u].push_back(val[u]);
        if (keep) tree.update(val[u],1);
        return;
    }
    int lt = adj[u][0], rt = adj[u][1];
    long long tmp = 0;
    dfs(lt,1);
    dfs(rt,1);
//    if (sz[lt] < sz[rt]) {
//        dfs(lt,0);
//        dfs(rt,1);
//        ch[u].swap(ch[rt]);
//        for (int i=0;i<ch[lt].size();i++) {
//            int v = ch[lt][i];
//            ch[u].push_back(v);
////            cerr << u << ' ' << v << '\n';
//            tmp += tree.query(1,v-1);
//        }
//        if (!keep) {
//            for (int i=0;i<ch[rt].size();i++) tree.update(ch[rt][i],-1);
//        }
//        else {
//            for (int i=0;i<ch[lt].size();i++) tree.update(ch[lt][i],1);
//        }
//    }
//    else {
//        dfs(rt,0);
//        dfs(lt,1);
//        ch[u].swap(ch[lt]);
//        for (int i=0;i<ch[rt].size();i++) {
//            int v = ch[rt][i];
////            cerr << u << ' ' << v << '\n';
//            ch[u].push_back(v);
//            tmp += tree.query(v+1,n);
//        }
//        if (!keep) {
//            for (int i=0;i<ch[lt].size();i++) tree.update(ch[lt][i],-1);
//        }
//        else {
//            for (int i=0;i<ch[rt].size();i++) tree.update(ch[rt][i],1);
//        }
//    }
    tmp = min(tmp,1ll * sz[lt] * sz[rt] - tmp);
    ans += tmp;
}

int main() {
    cin.tie(0) -> sync_with_stdio(0);
    cin >> n;
    tree.init(n);
    for (int i=1;;i++) {
        cin >> val[i];
        if (st.size()) adj[st.top().first].push_back(i);
        if (val[i] != 0) cnt++;
        if (val[i] != 0 && st.top().second == 2) st.pop();
        else if (val[i] != 0) st.top().second++;
        else {
            st.push({i,1});
        }
        if (cnt == n) break;
    }
    cal_sz(1);
    dfs(1,1);
    cout << ans;
    return 0;
}

Subtask #1
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 44 ms 55624 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #2
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 39 ms 55636 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 47 ms 55900 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #4
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 40 ms 56148 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #5
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 44 ms 56668 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #6
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 50 ms 60144 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #7
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 60 ms 62792 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #8
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 274 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #9
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 262 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #10
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 249 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #11
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 248 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -