ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

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

2021-07-12 18:03:26  阅读:215  来源: 互联网

标签:pq stcm int siz void LG7124 functor son Ynoi2008


传送门

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

假设当前递归到处理 \(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();}

这他妈都什么语法糖啊(

标签:pq,stcm,int,siz,void,LG7124,functor,son,Ynoi2008
来源: https://www.cnblogs.com/AThousandMoons/p/15003099.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有