CF1092E Minimal Diameter Forest

Love_xyh發表於2020-09-28
和CF455C Civilization貌似挺像。
我們求出每棵樹的直徑和中點,找到直徑最大的樹後,將別的樹的中點與直徑最大的樹的中點相連,即可完成構造。
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,m,u,v,ans,p,mid,pos;
int f[N];
bool vis[N],jay;
int cnt,head[N];
struct edge{int next,to;}e[N<<1];
int id,tot;
struct number{int len,id;}num[N];

inline void add(int u,int v)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	head[u]=cnt;
}

void dfs(int u,int fa,int now)
{
	vis[u]=true;
	if (ans<now) ans=now,p=u;
	for (register int i=head[u]; i; i=e[i].next) if (e[i].to!=fa) dfs(e[i].to,u,now+1);
}

void dfs2(int u,int fa)
{
	f[u]=1;
	for (register int i=head[u]; i; i=e[i].next) 
	if (e[i].to!=fa)
	{
		dfs2(e[i].to,u);
		f[u]=max(f[u],f[e[i].to]+1);
	}
}

void dfs3(int u,int fa,int now)
{
	if (now==mid) {pos=u; jay=true; return;}
	for (register int i=head[u]; i; i=e[i].next)
	if (e[i].to!=fa && f[e[i].to]+1==f[u]) 
	{
		dfs3(e[i].to,u,now+1);
		if (jay) return;
	}
}

int main(){
	scanf("%d%d",&n,&m);
	for (register int i=1; i<=m; ++i) scanf("%d%d",&u,&v),add(u,v),add(v,u);
	
	for (register int i=1; i<=n; ++i)
	if (!vis[i])
	{
		ans=-1; pos=i; p=i;
		dfs(i,0,0);
		dfs2(p,0);
		jay=false;
		mid=(f[p]+1)/2;
		dfs3(p,0,1);
		tot++; num[tot].len=f[p]; num[tot].id=pos;
	}
	id=1;
	for (register int i=2; i<=tot; ++i) if (num[i].len>num[id].len) id=i;
	for (register int i=1; i<=tot; ++i) if (i!=id) add(num[id].id,num[i].id),add(num[i].id,num[id].id);
	ans=0; p=0; dfs(1,0,0);
	ans=0; dfs(p,0,0);
	printf("%d\n",ans); 
	for (register int i=1; i<=tot; ++i) if (i!=id) printf("%d %d\n",num[id].id,num[i].id);
return 0;
}

相關文章