问题描述
现在有一棵二叉树,所有非叶子节点都有两个孩子。在每个叶子节点上有一个权值(有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;
}