洛谷 P11962[GESP6]树上漫步

P11962 [GESP202503 六级] 树上漫步

题目描述

小 A 有一棵 n 个结点的树，这些结点依次以 1,2,⋯,n 标号。

小 A 想在这棵树上漫步。具体来说，小 A 会从树上的某个结点出发，每一步可以移动到与当前结点相邻的结点，并且小 A 只会在偶数步（可以是零步）后结束漫步。

现在小 A 想知道，对于树上的每个结点，从这个结点出发开始漫步，经过偶数步能结束漫步的结点有多少个（可以经过重复的节点）。

输入格式

第一行，一个正整数 n。

接下来 n−1 行，每行两个整数 ui​,vi​，表示树上有⼀条连接结点 ui​ 和结点 vi​ 的边。

输出格式

一行，n 个整数。第 i 个整数表示从结点 i 出发开始漫步，能结束漫步的结点数量。

输入输出样例

输入 #1

3
1 3
2 3

输出 #1

2 2 1

输入 #2

4
1 3
3 2
4 3

输出 #2

3 3 1 3

思路

根据题意，不难发现如果将结点染色，相邻的结点染成不同的两种颜色，本题又要求走偶数步，即为求从一个结点出发，所能到达的颜色相同的节点数。

证明过程略。BFS一次即可。

代码

#include<iostream>
#include<vector>
using namespace std;

int main() {
	int n;
	cin>>n;
	vector<vector<int> > g(n+1);
	for(int i=0;i<n-1;i++) {
		int u,v;
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	
	vector<int> q={1};
	vector<int> col(n+1,-1);//结点的颜色
	int cnt0=0,cnt1=0;//cnt0为颜色是0的结点个数，cnt1为颜色是1的结点个数
	
	col[1]=0;
	for(int i=0;i<q.size();i++) {
		int u=q[i];
		if(col[u]==0) ++cnt0;
		else ++cnt1;
		for(int v:g[u]) if(col[v]==-1) {//未被访问过
			col[v]=col[u]^1;//将v染成与u不同的颜色
			q.push_back(v);
		}
	}
	
	for(int u=1;u<=n;u++)
		cout<<(col[u]==0?cnt0:cnt1)<<" \n"[u==n];
	return 0;
}