求本题原题或做法

RT。学校考试的，按照惯例，应该是搬的题。

有一棵树以及 $a_0,a_1…a_{n-1}$ 这个数组，每次可以选择一个点，然后把所有距离它为 $i$ 的点打上标记，花费为 $a_i$，求所有点都被打上标记的代价。