【luogu3174】【HAOI2009】毛毛虫

Description

对于一棵树，我们可以将某条链和与该链相连的边抽出来，看上去就象成一个毛毛虫，点数越多，毛毛虫就越大。

Input

在文本文件 worm.in 中第一行两个整数 N ， M ，分别表示树中结点个数和树的边数。
接下来 M 行，每行两个整数 a, b 表示点 a 和点 b 有边连接（ a, b ≤ N ）。你可以假定没有一对相同的 (a, b) 会出现一次以上。

Output

在文本文件 worm.out 中写入一个整数 , 表示最大的毛毛虫的大小。

Sample Input

13 12
1 2
1 5
1 6
3 2
4 2
5 7
5 8
7 9
7 10
7 11
8 12
8 13

Sample Output

11

Hint

40% 的数据， N ≤ 50000

100% 的数据， N ≤ 300000

Solution

维护包含该点的子树内最大毛毛虫，答案统计时为子树内最大毛毛虫+次大毛毛虫的结果。时间复杂度为O(n).

Code

#include <stdio.h>
#define MN 300005
#define R register
#define file(x) freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);
#define end fclose(stdin);fclose(stdout)
inline int read(){
	R int x; R bool f; R char c;
	for (f=0; (c=getchar())<'0'||c>'9'; f=c=='-');
	for (x=c-'0'; (c=getchar())>='0'&&c<='9'; x=(x<<3)+(x<<1)+c-'0');
	return f?-x:x;
}
int f[MN],ans,n,to[MN<<1],nt[MN<<1],h[MN],en,sz[MN];
inline void ins(int x,int y){to[++en]=y,nt[en]=h[x],h[x]=en;++sz[x];}
inline int max(int a,int b){return a>b?a:b;}
inline void dfs(int u,int fa){
	R int ma1=0,ma2=0;f[u]=1;
	for (R int i=h[u]; i; i=nt[i])
		if (to[i]!=fa){
			dfs(to[i],u);
			if (f[to[i]]>ma1) ma2=ma1,ma1=f[to[i]];
			else if (f[to[i]]>ma2) ma2=f[to[i]];
		}f[u]=max(f[u],ma1+sz[u]-1);ans=max(ans,ma2+ma1+sz[u]-1);
}
int main(){
	n=read(),read();
	for (R int i=1; i<n; ++i){
		R int x=read(),y=read();
		ins(x,y);ins(y,x);
	}dfs(1,0);printf("%d\n",ans);
}