BZOJ 2212 [Poi2011]Tree Rotations 线段树合并

题目链接

题意

现在有一棵二叉树，所有非叶子节点都有两个孩子。在每个叶子节点上有一个权值(有 $n$ 个叶子节点，满足这些权值为 $1..n$ 的一个排列)。可以任意交换每个非叶子节点的左右孩子。
要求进行一系列交换，使得最终所有叶子节点的权值按照遍历序写出来，逆序对个数最少。

参考

线段树的合并 ——杭州二中 黄嘉泰
【bzoj2212】[Poi2011]Tree Rotations ——hzwer // 基本上算是抄了一遍…。

Code

#include <bits/stdc++.h>
#define maxn 400010
using namespace std;
typedef long long LL;
LL ans, cnt1, cnt2;
int sum[maxn*10], ls[maxn*10], rs[maxn*10], root[maxn], v[maxn], sz, seg, l[maxn], r[maxn];
void read(int x) {
    scanf("%d", &v[x]);
    if (v[x]) return;
    l[x] = ++sz;
    read(l[x]);
    r[x] = ++sz;
    read(r[x]);
}
void push_up(int k) { sum[k] = sum[ls[k]] + sum[rs[k]]; }
void build(int& k, int l, int r, int val) {
    if (!k) k = ++seg;
    if (l == r) { ++sum[k]; return; }
    int mid = l + r >> 1;
    if (val <= mid) build(ls[k], l, mid, val);
    else build(rs[k], mid+1, r, val);
    push_up(k);
}
int merge(int x, int y) {
    if (!x) return y;
    if (!y) return x;
    cnt1 += 1LL * sum[ls[x]] * sum[rs[y]];
    cnt2 += 1LL * sum[rs[x]] * sum[ls[y]];
    ls[x] = merge(ls[x], ls[y]);
    rs[x] = merge(rs[x], rs[y]);
    push_up(x);
    return x;
}
void solve(int k) {
    if (!k) return;
    solve(l[k]), solve(r[k]);
    if (root[k]) return;
    cnt1 = cnt2 = 0;
    root[k] = merge(root[l[k]], root[r[k]]);
    ans += min(cnt1, cnt2);
}
void print(int x) {
    if (!x) return;
    printf("%d\n", x);
    print(l[x]), print(r[x]);
}
int main() {
    int n;
    scanf("%d", &n);
    read(++sz);
    for (int i = 1; i <= sz; ++i) {
        if (v[i]) build(root[i], 1, n, v[i]);
    }
    solve(1);
    printf("%lld\n", ans);
    return 0;
}