【bzoj2212&3702】二叉树

线段树合并入门题。

分别计算左子树的逆序对，右子树的逆序对，合并的时候计算贡献。

 #include<bits/stdc++.h>
 #define N 8000005
 using namespace std;
 typedef long long ll;
 int n,sz,tot,cnt=;
 ll cnt1,cnt2,ans;
 int val[N],ls[N],rs[N],rt[N],sumv[N],l[N],r[N];
 inline int read(){
     int f=,x=;char ch;
     do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
     do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
     return f*x;
 }
 void intree(int x){
     val[x]=read();
     if(!val[x]){
         l[x]=++cnt;intree(l[x]);
         r[x]=++cnt;intree(r[x]);
     }
 }
 inline void pushup(int o){sumv[o]=sumv[ls[o]]+sumv[rs[o]];}
 void build(int &o,int l,int r,int v){
     if(!o)o=++cnt;if(l==r){sumv[o]=;return;}
     int mid=(l+r)>>;
     if(v<=mid)build(ls[o],l,mid,v);
     else build(rs[o],mid+,r,v);
     pushup(o);
 }
 int merge(int x,int y){
     if(!x)return y;if(!y)return x;
     cnt1+=(ll)sumv[rs[x]]*sumv[ls[y]];
     cnt2+=(ll)sumv[ls[x]]*sumv[rs[y]];
     ls[x]=merge(ls[x],ls[y]);
     rs[x]=merge(rs[x],rs[y]);
     pushup(x);return x;
 }
 void solve(int x){
     if(!x)return;
     solve(l[x]);solve(r[x]);
     if(!val[x]){
         cnt1=cnt2=;
         rt[x]=merge(rt[l[x]],rt[r[x]]);
         ans+=min(cnt1,cnt2);
     }
 }
 int main(){
     n=read();cnt++;
     intree();
     for(int i=;i<=cnt;i++)if(val[i])build(rt[i],,n,val[i]);
     solve();
     printf("%lld\n",ans);
     return ;
 }