这题可以用虚树做吗？对于每个组别都建出虚树再做树形DP求直径。