如果您统计路径时是这样写的
ans += 2ll * sz[u] * (n-sz[u]) * a[u];
请改为
ans += 2ll * sz[u] * (SZ-sz[u]) * a[u];
其中 SZSZSZ 为此连通块的大小 (题目不保证图全部联通)