LG7124 [Ynoi2008] stcm【树分治，构造】

传送门

大概是全局平衡二叉树的思想（

假设当前递归到处理 (x) 的子树，首先我们先把以 (x) 为链顶的重链上的节点做了，因为这些集合互相包含，可以全部加一次再删掉。

接着要递归到轻子树，首先把重链节点加入，然后对轻子树建 Huffman 树，“加左进右，加右进左”，就可以对每个轻子树都把它以外的节点加入，递归到轻儿子。

加点操作次数是 (2.5nlog n)，可以过。

#include<bits/stdc++.h>
#define PB emplace_back
#define fi first
#define se second 
using namespace std;
template<typename Functor>
struct wrapper_type
{
    Functor functor;
    template<typename... Args>
    decltype(auto) operator()(Args&&... args) const&
    {
        return functor(functor, std::forward<Args>(args)...);
    }
};
template<typename Functor>
wrapper_type<typename std::decay<Functor>::type> wrapper(Functor&& functor)
{
    return{ std::forward<Functor>(functor) };
}
typedef pair<int, int> pii;
const int N = 100003;
int n, fa[N], son[N], siz[N];
vector<int> E[N];
void dfs(int x){
	siz[x] = 1;
	for(int v : E[x]){
		dfs(v); siz[x] += siz[v];
		if(siz[son[x]] < siz[v]) son[x] = v;
	}
}
void add(int x){printf("+%d", x); for(int v : E[x]) add(v);}
void rem(int x){while(x --) putchar('-');}
void work(int x){
	priority_queue<pii, vector<pii>, greater<pii> > pq;
	int len = 0;
	for(int u = x;u;u = son[u]){
		printf("=%d+%d", u, u); ++len;
		for(int v : E[u]) if(v != son[u]){
			add(v); pq.emplace(siz[v], v);
		}
	} rem(siz[x]);
	if(pq.empty()) return;
	for(int u = x;u;u = son[u]) printf("+%d", u);
	vector<int> sum, ls, rs;
	while(pq.size() > 1){
		pii L = pq.top(); pq.pop();
		pii R = pq.top(); pq.pop();
		sum.PB(L.fi + R.fi); ls.PB(L.se); rs.PB(R.se);
		pq.emplace(L.fi + R.fi, n + sum.size());
	}
	auto S = [&](int x){return x <= n ? siz[x] : sum[x-n-1];};
	auto Add = wrapper([&](auto&& self, int x) -> void{
		if(x <= n) add(x);
		else {x -= n+1; self(self, ls[x]); self(self, rs[x]);}
	});
	auto Work = wrapper([&](auto&& self, int x) -> void{
		if(x <= n) work(x);
		else { x -= n+1;
			Add(ls[x]); self(self, rs[x]); rem(S(ls[x]));
			Add(rs[x]); self(self, ls[x]); rem(S(rs[x]));
		}
	});
	Work(pq.top().se); rem(len);
}
void solve(){
	for(int i = 1;i <= n;++ i) E[i].resize(0);
	memset(son, 0, n+1<<2);
	for(int i = 2;i <= n;++ i){
		scanf("%d", fa+i);
		E[fa[i]].PB(i);
	} dfs(1); work(1); puts("!");
}
int main(){while(~scanf("%d", &n)) solve();}

这他妈都什么语法糖啊（