SP1437 PT07Z - Longest path in a tree

SP1437 PT07Z - Longest path in a tree

洛谷传送门

题意翻译

题目描述 给你一个无权无向的树。编写程序以输出该树中最长路径（从一个节点到另一个节点）的长度。在这种情况下，路径的长度是我们从开始到目的地的遍历边数。

输入输出格式 输入格式： 输入文件的第一行包含一个整数N ---树中的节点数（0 < N <= 10000）。接下来N -1行包含该树的N -1个边---每行包含一对（u，v），表示在节点u和节点v之间存在边（1 <= u，v <= N）。

输出格式： 一行：输出最长路径的长度。

---

题解：

树的直径模板。

代码：

#include<cstdio>
using namespace std;
const int maxn=10010;
int n;
int tot,head[maxn],nxt[maxn<<1],to[maxn<<1];
int maxx,pos,ans;
int deep[maxn];
void add(int x,int y)
{
	to[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
}
void dfs(int x,int f)
{
	deep[x]=deep[f]+1;
	if(deep[x]>maxx)
	{
		maxx=deep[x];
		pos=x;
	}
	for(int i=head[x];i;i=nxt[i])
	{
		int y=to[i];
		if(y==f)
			continue;
		dfs(y,x);
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	deep[0]=-1;
	dfs(1,0);
	maxx=0;
	dfs(pos,0);
	ans+=maxx;
	printf("%d\n",ans);
	return 0;
}