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;
}