学习 http://www.docin.com/p-598829500.html 线段树合并中的一道例题。
详细见那个课件。
复杂度$O(nlgn)$.
需要注意下空间

code:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
int n;
int LS[400005],RS[400005];
int cc;
int Root;
int w[400005];
int siz[400005];
int ls[15000500],rs[15000500],sum[15000500];
int num;
int stk[15000500],top;
int getnew(){
	if(top>0){
		ls[stk[top-1]]=rs[stk[top-1]]=sum[stk[top-1]]=0;
		return stk[--top];
	}
	return ++num;
}
void add(int &o,int l,int r,int pos){
	if(!o) o=getnew();
	if(l==r){
		sum[o]=1;
	}else{
		int mid=(l+r)>>1;
		if(pos<=mid) add(ls[o],l,mid,pos);else add(rs[o],mid+1,r,pos);
		sum[o]=1;
	}
}
int read(){
	int x;
	scanf("%d",&x);
	int now=++cc;
	if(!x){
		LS[now]=read();
		RS[now]=read();
		siz[now]=siz[LS[now]]+siz[RS[now]]+1;
	}else{
		add(w[now],1,n,x);
		siz[now]=1;
	}
	return now;
}
LL ans;
LL ans0,ans1;
int merge(int a,int b,int l,int r){
	if(!a) return b;
	if(!b) return a;
	int now=getnew();
	int mid=(l+r)>>1;
	int lft=sum[ls[a]]+sum[ls[b]];
	int rgt=sum[rs[a]]+sum[rs[b]];
	ans0+=sum[rs[a]]*1LL*sum[ls[b]];
	ans1+=sum[ls[a]]*1LL*sum[rs[b]];
	if(lft<rgt){
		rs[now]=merge(rs[a],rs[b],mid+1,r);
		ls[now]=merge(ls[a],ls[b],l,mid);
	}else{
		ls[now]=merge(ls[a],ls[b],l,mid);
		rs[now]=merge(rs[a],rs[b],mid+1,r);
	}
	stk[top++]=a;
	stk[top++]=b;
	sum[now]=sum[ls[now]]+sum[rs[now]];
	return now;
}
void dfs(int x){
	if(!LS[x]) return;
	else{
		if(siz[LS[x]]<siz[RS[x]]) dfs(RS[x]),dfs(LS[x]);
		else dfs(LS[x]),dfs(RS[x]);
		ans0=ans1=0;
		w[x]=merge(w[LS[x]],w[RS[x]],1,n);
		ans+=min(ans0,ans1);
	}
}
int main(){
	http://www.docin.com/p-598829500.html
	scanf("%d",&n);
	Root=read();
	dfs(1);
	printf("%lld\n",ans);
	return 0;
}