bzoj2212 Tree Rotations (线段树合并)

问题描述

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

input

第一行n
下面每行，一个数x
如果x==0，表示这个节点非叶子节点，递归地向下读入其左孩子和右孩子的信息，
如果x!=0，表示这个节点是叶子节点，权值为x

1<=n<=200000

output

一行，最少逆序对个数

Sample Input

3
0
0
3
1
2

Sample Output

1

分析:

一个节点如果交换两颗子树，对节点外的树没有影响

为每个叶子节点建立一棵权值线段树，自下而上合并
合并线段树的时候,计算如果交换子树的逆序对数量，和如果不交换子树的逆序对数量
取两者的最小值

假设一个节点的左右子树编号为x,y
右子树值域为(mid+1)-r肯定比左子树l-mid大，
所以:
如果不交换至少有a[rc[x]]*a[lc[y]]对逆序对
如果交换至少有a[lc[x]]*a[rc[y]]对逆序对
但是这样没统计完全，还要考虑x,y的左右子树lc[x],rc[x],lc[y],rc[y]中的逆序对
继续递归处理直到叶节点

code:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
#define ll long long
const int maxm=1e7+5;
int lc[maxm],rc[maxm],a[maxm],cnt;
ll ans,c1,c2;
int n,t;
int build(int l,int r,int x){//建立一棵权值线段树,只有x代表的叶子节点有值
    int k=++cnt;
    a[k]=1;//省去pushup
    if(l!=r){
        int mid=(l+r)/2;
        if(x<=mid){//只建立一边,另一边为空
            lc[k]=build(l,mid,x);
        }else{
            rc[k]=build(mid+1,r,x);
        }
    }
    return k;
}
int merged(int x,int y){
    if(!x||!y)return x+y;//如果其中一个为空，返回非空的作为合并之后的节点
    //否则任选一个作为合并之后的节点
    c1+=(ll)a[rc[x]]*a[lc[y]];//如果不交换(A的右子树和B的左子树),这里的统计是不完全的，全部递归之后才是准确的
    c2+=(ll)a[lc[x]]*a[rc[y]];//如果交换(A的左子树和B的右子树)
    lc[x]=merged(lc[x],lc[y]);//左子树合并
    rc[x]=merged(rc[x],rc[y]);//右子树合并
    a[x]=a[lc[x]]+a[rc[x]];
    return x;
}
int dfs(){
    scanf("%d",&t);
    if(t){//如果是叶子节点
        return build(1,n,t);//建一棵权值线段树
    }else{//如果不是叶子节点
        int l=dfs();//左子树
        int r=dfs();//右子树
        int node=merged(1,n,l,r);//合并两颗子权值线段树,并且求出交换 和不交换的逆序对数量
        ans+=min(c1,c2);//选择较少的那个
        c1=c2=0;//清空
        return node;
    }
}
signed main(){
    scanf("%d",&n);
    dfs();
    printf("%lld\n",ans);
    return 0;
}