bzoj 2212（线段树合并）

传送门
题解：权值线段树从下往上合并，每个点统计两个儿子中某一个交换/不交换的答案，取min后加到总答案中。不禁让人想起CDQZ Challenge 13
P.S.形态树开两倍空间，因为这种读入方式肯定会读成一棵满二叉树（不管存不存在子节点先加了时间戳）。另外，定义变量要搞清楚哪个是输入的形态树，哪个是线段树。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2e5+4;
int st,n,a[MAXN<<1],son[2][MAXN<<1],tim=0,tot=0;
int root[MAXN<<1],lc[MAXN*19],rc[MAXN*19],siz[MAXN*19];
ll t1,t2,ans=0;
inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
void init(int &x) {
    x=++tim;
    a[x]=read();
    if (!a[x]) init(son[0][x]),init(son[1][x]);
}
inline void pushup(int rt) {
    siz[rt]=siz[lc[rt]]+siz[rc[rt]];
}
void insert(int &rt,int l,int r,int val) {
    if (!rt) rt=++tot;
    if (l==r) {siz[rt]=1;return ;}
    int mid=(l+r)>>1;
    if (val<=mid) insert(lc[rt],l,mid,val);
    else insert(rc[rt],mid+1,r,val);
    pushup(rt);
}
int merge(int x,int y) {
    if (!x) return y;
    if (!y) return x;
    t1+=(ll)siz[lc[x]]*siz[rc[y]];
    t2+=(ll)siz[rc[x]]*siz[lc[y]];
    lc[x]=merge(lc[x],lc[y]);
    rc[x]=merge(rc[x],rc[y]);
    pushup(x);
    return x;
}
void work(int x) {
    if (a[x]) return ;
    work(son[0][x]),work(son[1][x]);
    t1=t2=0;
    root[x]=merge(root[son[0][x]],root[son[1][x]]);
    ans+=min(t1,t2);
}
int main() {
//  freopen("bzoj 2212.in","r",stdin);
    n=read();
    init(st);
    for (register int i=1;i<=tim;++i)
        if (a[i]) insert(root[i],1,n,a[i]);
    work(st);
    printf("%lld\n",ans);
    return 0;
}