bzoj 2212: [Poi2011]Tree Rotations 线段树合并

题意

现在有一棵二叉树，所有非叶子节点都有两个孩子。在每个叶子节点上有一个权值(有n个叶子节点，满足这些权值为1..n的一个排列)。可以任意交换每个非叶子节点的左右孩子。要求进行一系列交换，使得最终所有叶子节点的权值按照遍历序写出来，逆序对个数最少。
n<=200000

分析

对每个叶节点建立一棵线段树，然后每次合并的时候顺便求出交换孩子和不交换孩子的逆序对数，最后统计即可。

记得开long long

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 400005
#define ll long long
using namespace std;

int val[N],l[N],r[N],n,size,sz,root[N];
ll ans0,ans1,ans;
struct tree{int s,l,r;}t[N*20];

void init(int d)
{
    int x;
    scanf("%d",&x);
    if (x)
    {
        val[d]=x;
        return;
    }
    l[d]=++sz;r[d]=++sz;
    init(l[d]);init(r[d]);
}

void ins(int &d,int l,int r,int x)
{
    if (!d) d=++size;
    t[d].s++;
    if (l==r) return;
    int mid=(l+r)/2;
    if (x<=mid) ins(t[d].l,l,mid,x);
    else ins(t[d].r,mid+1,r,x);
}

int merge(int x,int y)
{
    if (!x) return y;
    if (!y) return x;
    ans0+=(ll)t[t[x].r].s*t[t[y].l].s;
    ans1+=(ll)t[t[x].l].s*t[t[y].r].s;
    t[x].l=merge(t[x].l,t[y].l);
    t[x].r=merge(t[x].r,t[y].r);
    t[x].s=t[t[x].l].s+t[t[x].r].s;
    return x;
}

void dfs(int d)
{
    if (!l[d]) ins(root[d],1,n,val[d]);
    else
    {
        dfs(l[d]);dfs(r[d]);
        ans0=ans1=0;
        root[d]=merge(root[l[d]],root[r[d]]);
        ans+=min(ans0,ans1);
    }
}

int main()
{
    scanf("%d",&n);
    sz=1;
    init(1);
    dfs(1);
    printf("%lld",ans);
    return 0;
}