H - Tourist Agency （ dfs ）

H - Tourist Agency （ dfs ）

题意：n个节点，n-1条边的生成树。选k个相连的点（ 1<=k<=n ），求可以选择的不同的种类数，和所有种类数的大小和。

比如：3个点E(1,2),E(1,3), 种类数有6种分别是(1),(2),(3),(1 2),(1 3),(1 2 3),。种类数的大小和是10=1+1+1+2+2+3

思路：先考虑求第一个答案，定义sum[i] 表示选择 i 节点及其子树可以有多少种类。叶子节点sum[]=1， 对于非叶子节点sum[i] = 子树的sum[]+1 的乘积。比如2号节点有两个儿子，一个儿子有1种，一个儿子有2种，sum[2]的定义是选择当前节点2，左儿子3我可以选也可以不选算2种，右儿子5可以选算2种不选算1种，一共3种情况，所以sum[2] = 2*3=6;

所有节点的种类数加起来就是答案一。

答案二就是统计，所有节点被选中了几次，sum[4]=7，选择4号节点的种类数有7种，所以4被选中了7次，答案贡献是7.

对于非根节点，显然不能这么算，因为他们除了儿子贡献还有来自父亲的贡献。可以对于每一个点跑一边答案一的dfs，这样就保证了每一个点都没有父亲节点，算出来的就是对答案二的贡献。但显然会超时，应该考虑在一个dfs中维护。

我们考虑在dfs2时，父亲顺便传到儿子一个贡献，代表父亲对他的影响。对于节点2，sum[3]=1, sum[5]=2, 我们再把2的父亲拿过来当作儿子，所以4节点传过来的值，应该是当2是根节点，4这个儿子的sum，这个sum是1。

再看节点5，考虑2应该传过来的值。想象5是根节点，2是其一个儿子，2的值应该是除去5这个父亲其他儿子的乘积，是2*2=4。

全部算完后加起来就是答案二。

 

#include
#define int long long

using namespace std;

const int mod = 1e9+7;
const int maxn = 2e5+10;
vector G[maxn];
int sum[maxn],sum2[maxn],n;

int qpow( int a, int n )
{
    int re = 1;
    while ( n ) {
        if ( n&1 ) re=(re*a)%mod;
        a=(a*a)%mod;
        n>>=1;
    }
    return re;
}

void dfs( int u, int f )
{
    sum[u] = 1;
    for ( int i=0; i>n;
    for ( int i=0; i