做题记录 牛客小白月赛46 D

 

 思路：dp,设dp_u,i代表以u为根，距离为i+1的节点总数。

最后dp_u,1要加上所有dp_v,0-1，其中v是u的子节点（要-1是为了排除v->u的边）

则每个节点i的答案为dp_i,0+dp_i,1+1

代码如下:

    for(int i=1;i<=n;i++){
        for(int j=fir[i]; ~j; j=G[j].nex) {
            if(G[j].wei==1)        dp[i][0]++;
            elsif(G[j].wei==2)     dp[i][1]++;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=fir[i]; ~j; j=G[j].nex) {
            int to=G[j].to;
            if(G[j].wei==1)    dp[i][1]+=(dp[to][0]-1); //排除j->i的边
        }
    }