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];
}