赛时想到的是，最优路径不可能经过$-1$，于是考虑设 $dp_u$ 作为一个 vector 表示以 $u$ 为根的子树中从 $u$ 开始的最优路径上 $a_j = 1$ 的所有 $j$，那么转移的时候只需要从前往后对比两个 vector 的 $j$ 的 $dep$ 的大小进行转移。

然后这个做法 mle 了，于是考虑了一种优化，设 $dp_u$ 表示最优路径上第一个 $a_j=1$ 的 $j$，然后转移的时候同样和对比 vector 一样，只不过空间复杂度得到优化，但是这样做的时间复杂度是正确的吗？

代码