[bzoj2212]Tree Rotations

初始对于每一个叶子节点建一棵权值线段树，然后再线段树合并时计算一下两者分别在前面和后面的代价，$ans+=min(\sum sz[rz[x]]\cdot sz[ls[y]],\sum sz[ls[x]]\cdot sz[rs[y]])$

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 200001
#define mid (l+r>>1)
int V,n,ls[N*20],rs[N*20];
ll ans,ans1,ans2,sz[N*20];
void update(int k,int l,int r,int x){
    sz[k]=1;
    if (l==r)return;
    if (x<=mid)update(ls[k]=++V,l,mid,x);
    else update(rs[k]=++V,mid+1,r,x);
}
int merge(int x,int y){
    if (x*y==0)return x+y;
    if (ls[x]+rs[x]==0)sz[x]+=sz[y];
    else{
        ans1+=sz[ls[x]]*sz[rs[y]];
        ans2+=sz[rs[x]]*sz[ls[y]];
        sz[x]=sz[ls[x]=merge(ls[x],ls[y])]+sz[rs[x]=merge(rs[x],rs[y])];
    }
    return x;
}
int dfs(){
    int x,r,r1=0,r2=0;
    scanf("%d",&x);
    if (x){
        update(r=++V,1,n,x);
        return r;
    }
    r1=dfs();
    r2=dfs();
    ans1=ans2=0;
    r=merge(r1,r2);
    ans+=min(ans1,ans2);
    return r;
}
int main(){
    scanf("%d",&n);
    dfs();
    printf("%lld",ans);
}