题意有点绕,其实就是让你求一个点能被多少个点集包含,同时这些点集内的点要相互连通。
首先,简单来说,如果只是计算一个有根树中任意一个点被多少个只包含它以及它的子树的点的点集包含,那么直接普通的树上统计的trick就可以搞定。但是现在问题是,点集的点可以是其子树之外的点。我们注意到,对于根来说,它的答案都满足点集里面的点都在它的子树里面,所以说如果能够把每一个点都当作根来计算一次,那么就可以简化问题。
我们令f[x]表示包含x而且点集里面的点都在x的子树里面的点集的个数。那么f[x]可以通过迭代以下式子得出:
可以看到,这个式子最后可以化简成x的儿子对应点集数加一的乘积,具体来说: