[BZOJ2212][POI2011]ROT-Tree Rotations（线段树合并）

题目：

我是超链接

题解：

统计一棵树的逆序对数就是左子树逆序对数+右子树逆序对数+左右子树之间的逆序对数
这个左右子树之间的话，如果我们用权值线段树，那么可以通过讨论交不交换，用左子树个数*右子树之类的，然后直接求min值
注意要区分原树中的左右节点和线段树中的左右节点

代码：

#include <cstdio>
#include <iostream>
#define LL long long
using namespace std;
const int N=1000005;
int v[N],l[N],r[N],sz,zs,ls[N*4],rs[N*4],sum[N*4],root[N*4];LL ans,cnt1,cnt2;
void build(int now)
{
    scanf("%d",&v[now]);
    if (!v[now])
    {
        l[now]=++sz;
        build(l[now]);
        r[now]=++sz;
        build(r[now]);
    }
}
void insert(int &now,int l,int r,int x)
{
    sum[++zs]=sum[now]+1; now=zs;
    if (l==r) return;int mid=(l+r)>>1;
    if (x<=mid) insert(ls[now],l,mid,x);
    else insert(rs[now],mid+1,r,x);
}
int merge(int x,int y)
{
    if (!x || !y) return x+y;
    cnt1+=(LL)sum[rs[x]]*sum[ls[y]];//不交换 
    cnt2+=(LL)sum[rs[y]]*sum[ls[x]];
    ls[x]=merge(ls[x],ls[y]);
    rs[x]=merge(rs[x],rs[y]);
    sum[x]=sum[ls[x]]+sum[rs[x]];
    return x;
}
void solve(int now)
{
    if (!now) return;
    solve(l[now]); solve(r[now]);
    if (!v[now])
    {
        cnt1=cnt2=0;
        root[now]=merge(root[l[now]],root[r[now]]);
        ans+=min(cnt1,cnt2);
    }
}
int main()
{
    int n;scanf("%d",&n);sz=1;
    build(1);
    for (int i=1;i<=sz;i++)
      if (v[i]) insert(root[i],1,n,v[i]);
    solve(1);
    printf("%lld",ans);
}