题目:
题解:
统计一棵树的逆序对数就是左子树逆序对数+右子树逆序对数+左右子树之间的逆序对数
这个左右子树之间的话,如果我们用权值线段树,那么可以通过讨论交不交换,用左子树个数*右子树之类的,然后直接求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);
}