Description
现在有一棵二叉树,所有非叶子节点都有两个孩子。在每个叶子节点上有一个权值(有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
s
o
l
u
t
i
o
n
solution
solution:
裸的线段树合并
只要考虑左子树右子树互相的影响即可
#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back
#define pc putchar
#define mp make_pair
#define file(k) memset(k,0,sizeof(k))
#define ll long long
int rd()
{
int num = 0;char c = getchar();bool flag = true;
while(c < '0'||c > '9') {if(c == '-') flag = false;c = getchar();}
while(c >= '0' && c <= '9') num = num*10+c-48,c = getchar();
if(flag) return num;else return -num;
}
ll ans1,ans2;
int n,t,cnt;
int ch[401000][2];
int val[401000],rt[401000];
int ls[16001000],rs[16001000],num[16001000];
int tmp[201000],tot;
void init(int x)
{
int y = rd(); ch[x][0] = ++t;
if(!y) init(t);
else val[t] = y,tmp[++tot] = y;
y = rd(); ch[x][1] = ++t;
if(!y) init(t);
else val[t] = y,tmp[++tot] = y;
}
int merge(int x,int y)
{
if(!x) return y;
if(!y) return x;
int root = ++cnt;
ans1 += 1ll*num[ls[x]]*num[rs[y]];
ans2 += 1ll*num[ls[y]]*num[rs[x]];
ls[root] = merge(ls[x],ls[y]);
rs[root] = merge(rs[x],rs[y]);
num[root] = num[ls[root]] + num[rs[root]];
return root;
}
void add(int &root,int l,int r,int pl)
{
root = ++cnt;
num[root] = 1;
if(l == r) return;
int mid = l+r>>1;
if(pl <= mid) add(ls[root],l,mid,pl);
else add(rs[root],mid+1,r,pl);
}
ll work(int x)
{
ll ans = 0;
if(!val[x])
{
ans = work(ch[x][0]) + work(ch[x][1]);
ans1 = ans2 = 0;
rt[x] = merge(rt[ch[x][0]],rt[ch[x][1]]);
ans += min(ans1,ans2);
}
else add(rt[x],1,tot,val[x]);
return ans;
}
int main()
{
n = rd();
int x = rd();
if(x) {printf("0\n");return 0;}
else init(++t);
sort(tmp+1,tmp+tot+1);
tot = unique(tmp+1,tmp+tot+1) - tmp - 1;
rep(i,1,t) if(val[i]) val[i] = lower_bound(tmp+1,tmp+tot+1,val[i]) - tmp;
printf("%lld\n",work(1));
return 0;
}