P8025 【樹鏈剖分求祖先】

Fire_Raku發表於2024-04-04

P8025 【樹鏈剖分求祖先】

這題的題意簡單,簡單分類討論一下這裡就不贅述了。最後題目就簡化成求一個點的 \(k\) 級祖先。

開始會的解法是倍增,但是常數過高被卡了。

常數更優秀的做法是樹剖,每一次跳樹鏈,最後可能有一條鏈太長只能跳一部分,這是因為樹鏈剖分的 \(dfn\) 序是有序的,即每條鏈佔的是一段區間,所以直接查詢 \(dfn\) 陣列即可。

int jump(int u, int t) { //從 u 點往上跳 t 次
	while(t > 0) {
		if(t - (dep[u] - dep[top[u]] + 1) >= 0) t -= dep[u] - dep[top[u]] + 1, u = fa[top[u]];
		else break;
	}
	return wt[id[u] - t];
}

相關文章