Codeforces Beta Round #14 (Div. 2) D. Two Paths（树的直径）

题目链接


题意：给定一棵树，要你在树上找两条路径，使得这两条路径长度的乘积最大（两条路径长度不能有公共点），输出最大值。
思路：考虑一下每条边的贡献，如果割去一条边，那么就会形成两棵树，取这两棵树的直径再取max就行了。这里求直径的时候记得id要初始化。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+1;
vector<int>g[maxn];
int id,mx,a[maxn],b[maxn];
void dfs(int x,int fa,int y,int deep)
{
	if(deep>mx) mx=deep,id=x;
	for(auto to:g[x])
	{
		if(to==fa||to==y) continue;
		dfs(to,x,y,deep+1);
	}
}
int check(int a,int b)
{
	mx=0;
	dfs(a,a,b,0);
	mx=0;
	dfs(id,id,b,0);
	return mx;
}
int main()
{
	int n,u,v,ans=0;
	scanf("%d",&n);
	for(int i=1;i<n;++i)
	{
		scanf("%d%d",&a[i],&b[i]);
		g[a[i]].push_back(b[i]);
		g[b[i]].push_back(a[i]);
	}
	for(int i=1;i<n;++i)
	{
		id=a[i];
		int d1=check(a[i],b[i]);
		id=b[i];
		int d2=check(b[i],a[i]);
		ans=max(ans,d1*d2);
	}
	printf("%d\n",ans);
 }