#Z2294. 打印树的直径

Description

给你一棵树,树上有N个点,编号从0到N-1

请找出任意一条树的直径,并输出直径上的点,输出顺序为从直径的某个端点走向另一个端点

Format

Input

第一行一个整数 n;

之后 n-1 行每行两个整数 u,v,表示 u 和 v 之间有边。

1<=N<=2e5

Output

如题

Samples

输入数据 1

复制代码
10
0 1
0 2
0 4
0 6
0 7
1 3
2 5
4 8
6 9

Copy

输出数据 1

复制代码
3 1 0 2 5

思路

是树的直径的加难板,不会的可以看看求树的直径(史上最详细,匠心之作)_树的直径存在负边权-CSDN博客

我们可以在求完树的直径的两个端点后再做一次dfs并用栈储存路径即可,详解代码。


代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
vector<int>vec[1000001];
bool vis[10000001];
int ans,dep[1000001],n,x,y,len,t,tt,ttt,a[1000001],as;
stack<int> st;
void dfs(int nd,int deep)
{
  dep[nd] = deep;
  vis[nd] = 1;
  for(int i = 0;i < vec[nd].size();i++)
  {
    int son = vec[nd][i];
    if(vis[son] == 0)
    {
      dfs(son,deep + 1);
      
    }
  }
}
void dfs_2(int nd)
{
  vis[nd] = 1;
  st.push(nd);
  if(nd == ttt)
  {
    while(st.size())
    {
      a[++as] = st.top();
      st.pop();
    }
    for(int i = 1;i <= as;i++) cout<<a[i]<<' ';
    exit(0);
  }
  for(int i = 0;i < vec[nd].size();i++)
  {
    int son = vec[nd][i];
    if(vis[son] == 0) dfs_2(son);
  }
  st.pop();
}
signed main()
{
	cin>>n;
	for(int i = 1;i < n;i++)
	{
		cin>>x>>y;
		vec[x].push_back(y);
		vec[y].push_back(x);
	}
	dfs(0,1);
  for(int i = 0;i < n;i++)
    if(tt < dep[i])
    {
      tt = dep[i];
      t = i;
    }
  memset(vis,0,sizeof(vis));
  memset(dep,0,sizeof(dep));
  tt = 0;
  dfs(t,0);
  for(int i = 0;i < n;i++)
  {
    if(tt < dep[i])
    {
      tt = dep[i];
      ttt = i;
    }
  }
  //t~ttt
  memset(vis,0,sizeof(vis));
  dfs_2(t);
  return 0;
}
相关推荐
睡不醒的kun4 小时前
leetcode算法刷题的第三十二天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
_OP_CHEN6 小时前
数据结构(C语言篇):(十二)实现顺序结构二叉树——堆
c语言·数据结构·算法·二叉树·学习笔记··顺序结构二叉树
cellurw8 小时前
EDID 数据结构解析与编辑工具:校验和计算、厂商/设备名编解码、物理地址读写、颜色与时序信息提取
数据结构
Pluchon9 小时前
硅基计划3.0 Map类&Set类
java·开发语言·数据结构·算法·哈希算法·散列表
重生之我是Java开发战士10 小时前
【数据结构】Java集合框架:List与ArrayList
java·数据结构·list
爱干饭的boy10 小时前
手写Spring底层机制的实现【初始化IOC容器+依赖注入+BeanPostProcesson机制+AOP】
java·数据结构·后端·算法·spring
躲在云朵里`11 小时前
Redis深度解析:核心数据结构、线程模型与高频面试题
数据结构·数据库·redis
wangwangblog13 小时前
LLVM 数据结构简介
开发语言·数据结构·c++
平平无奇。。。16 小时前
解密完全二叉树顺序存储之堆结构
c语言·数据结构·visual studio