标签：结点 生活 int 距离 顶点 树上 dp

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

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。