首先\(S(u)\)显然是\(u\)的子树。假如\(u\)是定点，问题转化为区间求平方和，十分简单。于是我们用线段树维护区间平方和，支持区间加，然后离线问题，在\(u\)的位置处理即可。线段树从\(fa\)转移到\(u\)是极度简单的。时间复杂度\(O(n\logn)\)。#include<bits/stdc++.h>