P9304 「DTOI-5」3-1-树形数据结构

P9304 「DTOI-5」3-1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

//题意:给一棵树,每条边经过的时间为1,有一次从任意节点瞬间跳上根节点的机会(花费时间为0)。问走过i个节点且回到根节点的最小时间。
//要使花费的时间最小，就需要在最深的地方上跳
//如果你在一个很浅的节点上跳了，那么你想继续访问更深的节点显然就要跑得更深还要跑上来，显然不优。
//画个图，自己计算一下，就会找出规律
//我们先沿着最深的节点的那条路径走，这时每多访问一个点就会多走一条边。
//但是在这条路径之外，由于新走了一个点还要往回走回这条路径，所以会多走两条边。
//由此可以得出结论，分两种情况
//1.访问的节点数量小于最深深度的情况--->答案 =节点数量减一
//2.访问的节点数量大于最深深度的情况--->每多一个节点，多走两条边--->(max_dep-1)+(i-max_dep)*2

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
//题意:给一棵树,每条边经过的时间为1,有一次从任意节点瞬间跳上根节点的机会(花费时间为0)。问走过i个节点且回到根节点的最小时间。
//要使花费的时间最小，就需要在最深的地方上跳
//如果你在一个很浅的节点上跳了，那么你想继续访问更深的节点显然就要跑得更深还要跑上来，显然不优。
//画个图，自己计算一下，就会找出规律
//我们先沿着最深的节点的那条路径走，这时每多访问一个点就会多走一条边。
//但是在这条路径之外，由于新走了一个点还要往回走回这条路径，所以会多走两条边。
//由此可以得出结论，分两种情况
//1.访问的节点数量小于最深深度的情况--->答案 =节点数量减一
//2.访问的节点数量大于最深深度的情况--->每多一个节点，多走两条边--->(max_dep-1)+(i-max_dep)*2
using namespace std;
int n,max_dep;//max_dep是树中最大的深度
vector<int> q[100005];
int dfs(int cur,int fa){//从下向上计算最深的深度，先算子树再算根节点
	int dep=0;
	for(auto v:q[cur]){//遍历
		if(v==fa) continue;//不能是父节点
		dep=max(dep,dfs(v,cur));//每次取深度最大的子树深度
	}
	return dep+1;//加上当前层，即深度加1
}
void solve(){
	cin>>n;
	for(int i=0;i<n-1;i++){
		int u,v;
		cin>>u>>v;
		q[u].push_back(v);//存树
		q[v].push_back(u);
	}
	max_dep=dfs(1,0);//计算最大深度
	for(int i=1;i<=n;i++){
		//这个就是贪心了，可以画个图自己算一算，会找出来规律的，规律详情可看代码最上方
		if(i<=max_dep) cout<<i-1<<endl;
		else{
			cout<<(max_dep-1)+(i-max_dep)*2<<endl;
		}
	}
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	solve();
	return 0;
}