LG7124 [Ynoi2008] stcm【树分治,构造】
大概是全局平衡二叉树的思想(
假设当前递归到处理 \(x\) 的子树,首先我们先把以 \(x\) 为链顶的重链上的节点做了,因为这些集合互相包含,可以全部加一次再删掉。
接着要递归到轻子树,首先把重链节点加入,然后对轻子树建 Huffman 树,“加左进右,加右进左”,就可以对每个轻子树都把它以外的节点加入,递归到轻儿子。
加点操作次数是 \(2.5n\log 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();}
这他妈都什么语法糖啊(