[POI2011]ROT-Tree Rotations（线段树合并）

题意

一棵树，叶子有标号。

可以交换一个节点的两个子树，求一个叶子的前序遍历，使得逆序对个数最少。

思路

简单线段树合并。

对于一个非叶子节点只有两种可能：左儿子排前面或者右儿子排前面。全都算出来，取min就好了。然后合并两颗权值线段树。

然而我居然不会写merge然后T飞了？？？merge的常数应该不到1（假如你写一个query+一个update代替merge的话这题就苟不住了）。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define pLL pair<LL, LL>
#define fi first
#define se second
#define mp make_pair
const int N = 2e5 + 10, E = N * 21;
int n; LL ans;
int root[N], siz[N];
int ncnt, ls[E], rs[E], cnt[E];

void push_up(int u)
{
	cnt[u] = cnt[ls[u]] + cnt[rs[u]];
}

void add(int &u, int l, int r, int X, int Y)
{
	if (u == 0) u = ++ ncnt;
	if (l == r){
		cnt[u] += Y;
		return;
	}
	int mid = (l + r) >> 1;
	if (X <= mid) add(ls[u], l, mid, X, Y);
	else add(rs[u], mid+1, r, X, Y);
	push_up(u);
}

void pls(pLL &x, pLL y)
{
	x.fi += y.fi;
	x.se += y.se;
}

pLL merge(int &u, int s)
{
	if (!u || !s){
		u = u ^ s;
		return mp(0, 0);
	}
	pLL ret = mp(1LL * cnt[ls[u]] * cnt[rs[s]], 1LL * cnt[rs[u]] * cnt[ls[s]]);
	pls(ret, merge(ls[u], ls[s]));
	pls(ret, merge(rs[u], rs[s]));
	cnt[u] += cnt[s];
	return ret;
}

int solve()
{
	int u;
	scanf("%d", &u);
	if (u){
		add(root[u], 1, n, u, 1);
		siz[u] = 1;
	}
	else{
		int s; LL con = 0;
		u = solve();
		s = solve();
		if (siz[s] > siz[u]) swap(s, u);
		pLL ret = merge(root[u], root[s]);
		ans += min(ret.fi, ret.se);
		siz[u] += siz[s];
	}
	return u;
}

int main()
{
	scanf("%d", &n);
	solve();
	printf("%lld\n", ans);
	return 0;
}