距离为 $k$ 是点分树板子，考虑每个 dep 分别处理是不是就做完了，复杂度双 $log$，但是点分树常数巨大估计跑得巨慢，懒得打了，有没有人愿意打一下。