Codeforcesラウンド#615(DIV。3)木(ツリー径+ DFS)上の3つのパスF.

リンカスクリプトの追加
ここに画像を挿入説明
限り、私たちは木の直径で、ルート上の別のポイントをマークして、木の直径の2点である3点は2点を持っている必要があります結論、A、B、Cがあり、その後、トラバースマークをポイント:アイデアを最大距離の最大径の非標識ポイントプラス答えです。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+1;
vector<int>g[maxn];
int n,k,maxx=0,ans,first,second,father[maxn],Max=0,t,vis[maxn];
void dfs1(int u,int fa,int deep)//求树的直径
{
	if(deep>maxx)
	{
		maxx=deep;
		k=u;
	}
	for(int to:g[u])
	{
		if(to==fa) continue;
		father[to]=u;
		dfs1(to,u,deep+1);
	}
}
void dfs2(int u,int fa,int deep)
{
	if(deep>maxx)
	{
		maxx=deep;
		k=u;
	}
	for(int to:g[u])
	{
		if(to==fa) continue;
		if(vis[to]) continue;
		dfs2(to,u,deep+1);
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;++i){
		int u,v;
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs1(1,0,0);
	first=k;maxx=0;
	dfs1(k,0,0);
	second=k;
	ans=maxx;
	vis[first]=1;
	for(int i=second;i!=first;i=father[i]) {
		if(i!=second&&i!=first) t=i;
		vis[i]=1;
	}
	maxx=0;
	for(int i=1;i<=n;++i)
	if(vis[i]) {
		dfs2(i,0,0);
		if(maxx>Max) Max=maxx,t=k;
		maxx=0;
	}
	printf("%d\n",ans+Max);
	printf("%d %d %d\n",second,first,t);
 } 
公開された70元の記事 ウォンの賞賛0 ビュー2431

おすすめ

転載: blog.csdn.net/qq_42479630/article/details/104232892