D.生活在树上

思路：树上\(dp\),\(dp[i][1]\)表示从顶点\(i\)开始，走过的距离为\(1\)的经过的结点数，\(dp[i][2]\)表示从顶点\(i\)开始走过的距离为\(2\)经过的结点数。

初始化：统计一下每个结点周围的距离为\(1,2\)结点个数即可。

状态转移：\(if(dis == 1) dp[i][2] += dp[x][1] - 1\)。

int main()
{
	scanf("%d",&n);
	for(int i = 2; i <= n; i ++ ){
		int a, w;
		scanf("%d %d",&a, &w);
		v[a].pb({i,w}), v[i].pb({a,w});
	}
	for(int i = 1; i <= n; i ++ ){
		for(auto[x, dis]:v[i]){
			if(dis == 1) dp[i][1] ++;
			if(dis == 2) dp[i][2] ++;
		}
	}
	for(int i = 1; i <= n; i ++ ){
		for(auto [x, dis]: v[i]){
			if(dis == 1) dp[i][2] += dp[x][1] - 1;
		}
	}
	for(int i = 1; i <= n; i ++ ){
		printf("%d\n",dp[i][1] + dp[i][2] + 1);
	}
	return 0;
}

标签：结点,生活,int,距离,顶点,树上,dp
来源： https://www.cnblogs.com/acmerbs/p/16058434.html