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;
}