树链剖分

看逆天算法,品百味OI#4

1.重链剖分

给出如下定义:

  • 重子结点:一个点的子节点中子树最大的点

  • 轻子节点:剩余的子节点

  • 重边:该节点到重子节点的边

  • 轻边:到轻子节点的边

  • 重链:若干条首尾相接的重边

预处理由两个DFS实现

  • DFS1:记录节点的父节点、深度、子树大小、重子结点

  • DFS2:记录所在链的链顶,重边优先遍历时的DFS序、DFS序对应的节点编号

这样之后就可以把树上操作转化为区间操作(重链遍历下\(dfn\)序连续),可以使用线段树或树状数组维护

查询时用到类似\(LCA\)的思想:先把两个点跳到同一条链上,这时可以顺手对经过的区间\([dfn_{fa_x}],dfn_x]\)上来一波操作(求极值,求和等),跳完后再操作\([dfn_x,dfn_y]\)

板子:

P3178 [HAOI2015] 树上操作

「ZJOI 2008」树的统计

QTREE

用到了边权下放(\(dfs\)时实现),这种情况下\(lca\)的值不能纳入查询范围,便有了

ST.query(dfn[x] + 1,dfn[y],1,n,1)

(\(lca\)就是跳到一条链上后深度小的点)

[SDOI2011]染色

合并区间时要特判衔接处是否能合成一个区间,而且正是因为如此,线段树\(query\)的时候如果不跨\(mid\)就分开,只有跨了的时候才判一下,具体而言就是

	if(ql > mid) query(ql,qr,mid + 1,r,rs(id));
	else if(qr <= mid) query(ql,qr,l,mid,ls(id));
	else
	{
		ans = query(ql,qr,l,mid,ls(id)) + query(ql,qr,mid + 1,r,rs(id));
		if(t[ls(id)].rc == t[rs(id)].lc) ans--;
		return ans;
	}

全是细节

货车运输

没错就是最大生成树那个

原本的\(lca\)同时求\(min\)直接变成树剖后区间求\(min\),更好想了。码量也大了

运输计划

没错就是那个二分答案后check能否删掉超过答案的\(m\)条路径的边的交集中的边来达到要求的那个,拿lca+差分搞的那个

学了树剖后有了全新大发现

  • 沿用二分思路,用树剖求\(lca\),好像也没啥特别的还加了码量,就lca短了点,优化了常数

  • 考虑到树剖的区间操作优势,我们可以直接通过修改那\(m\)条路径来得到答案(正片)

我们选取\(m\)条路径中最长的那条作为标准,记为\(A\)

如果我们删掉了\(A\)中的一条边\(E\),那么最大值只有两种来源:删掉了边的\(A\)和不经过\(A\)的链中最长的

很好想,过\(E\)的其他路径权值和本来就比\(A\)要小,删了后绝对无优势,反而是那些不受影响的链可能趁机成为最大

这样一来又有重要的一点:删掉不同的边,结果极可能变化

也很好想,删掉不同的\(E\),不经过\(E\)的路径的集合也不同,变化多样

总结就是:

\(A\)中的所有边挨个删一遍,对于删掉\(E_i\)后的情况,求删掉了边的\(A\)和不经过\(A\)的路径中最长的路径,记为\(Max_i\),结果就是\(\min\limits_{E_i \in A}{(Max_i)}\)

对于一条路径,剖分后可以把它拆成若干条链(重/轻链的一部分),设为\((u_i,v_i)\),共\(k\)个,那么,这些区间的补集就是该路径所不经过的,即\((1,u_1),\cdots (v_{i}+1,u_{i + 1}-1 ),\cdots (v_k,n)\)

对于每一个小段,我们可以预处理出不过这一段的路径中最长的。从而在查询时方便一点

void init(int x,int y,int z)
{
	int res = 0;
	int fx = top[x],fy = top[y];
//	cout << fx << " " << fy << endl;
	while(fx != fy)
	{
		//cout << 114 << endl;;
		if(dep[fx] >= dep[fy]) {res++,P[res].c = dfn[fx],P[res].d = dfn[x];x = fa[fx];}
		else {res++,P[res].c = dfn[fy],P[res].d = dfn[y];y = fa[fy];}
		fx = top[x];fy = top[y];	
	}
	if(dfn[x] > dfn[y]) swap(x,y);
	res++;P[res].c = dfn[x] + 1,P[res].d = dfn[y];
	sort(P + 1,P + res + 1,cmp);
	if(P[1].c > 1) ST.add(1,P[1].c - 1,1,n,1,z);
	if(P[res].d < n) ST.add(P[res].d + 1,n,1,n,1,z);
	for(int i = 1;i < res;i++)
	{
	//	cout << P[i].d + 1 << " " << P[i + 1].c - 1 << endl;
		ST.add(P[i].d + 1,P[i + 1].c - 1,1,n,1,z);
	}
	//return;
}

时间效率(从高到低):树剖+ 补集(极限330ms)> 树剖+二分(极限约500ms) > 二分 + 倍增法lca(极限1.12s)

树剖常数真小

遥远的国度

换根问题

  • 查询节点就是当前根,查询整个树

  • 查询节点不是当前根的祖先,直接查询以该节点为根的子树

  • 查询节点是当前根的祖先,这时要按跳链法找到当前节点的儿子\(son\),那么\([dfn_{son},dfn_{son} + siz_{son} - 1]\)(以该儿子为根的子树)不纳入查询范围,即查询\([1,dfn_{son} - 1] \cup [dfn_{son} + siz_{son},n]\)

找儿子:

int look(int x)
{
	if(dfn[x] >= dfn[root] || dfn[x] + siz[x] - 1 < dfn[root]) return 114;
	int pos = root;
	while(top[pos] != top[x])
	{
		
		if(fa[top[pos]] == x) return top[pos];
		pos = fa[top[pos]];
	}
	return hson[x];
}

可以看到,树剖的魅力就是把树转成纯纯的区间,查询某一部分就是对某一区间进行操作,这样就可以用线段树来降复杂度,非常的有实力

posted @ 2024-02-17 12:24  why?123  阅读(13)  评论(0编辑  收藏  举报