题目描述
对于一棵树，我们可以将某条链和与该链相连的边抽出来，看上去就象成一个毛毛虫，点数越多，毛毛虫就越大。例如下图左边的树（图 1 ）抽出一部分就变成了右边的一个毛毛虫了（图 2 ）。
输入输出格式
输入格式：
在文本文件 worm.in 中第一行两个整数 N ， M ，分别表示树中结点个数和树的边数。
接下来 M 行，每行两个整数 a, b 表示点 a 和点 b 有边连接（ a, b ≤ N ）。你可以假定没有一对相同的 (a, b) 会出现一次以上。
输出格式：
在文本文件 worm.out 中写入一个整数 , 表示最大的毛毛虫的大小。
输入输出样例
输出样例#1：
11
说明
40% 的数据， N ≤ 50000
100% 的数据， N ≤ 300000
解题思路：树形DP
f[i]表示以i为根的子树的最长长度。
link[i]表示与i相连的点的个数
那么对于某个节点i，他对答案的贡献为:
由i扩展出去的最长链长度+由i扩展出去的次长链长度+与i相连的点的个数(注意更新时节点的重复)
#include<iostream>
#include<cstdio>
#include<algorithm>
#define DB double
#include<cmath>
#define pi 3.1415926535898
using namespace std;
inline int read()
{
int x=0,w=1;char ch=getchar();
while(!isdigit(ch)){if(ch==‘-‘) w=-1;ch=getchar();}
while(isdigit(ch)) x=(x<<3)+(x<<1)+ch-‘0‘,ch=getchar();
return x*w;
}
const int N=300009;
struct node{
int u,v,ne;
}e[N*2];
int n,h[N],tot;
void add(int u,int v)
{
tot++;e[tot]=(node){u,v,h[u]};h[u]=tot;
}
int m,link[N],f[N],ans;
void dfs(int x,int fa)
{
int l1=-1,l2=-1;
for(int i=h[x];i;i=e[i].ne)
{
int rr=e[i].v;
if(rr==fa) continue;
dfs(rr,x);
if(f[rr]>l2)
{
if(f[rr]>l1) l2=l1,l1=f[rr];
else l2=f[rr];
}
f[x]=max(f[x],f[rr]+link[x]-1);
}
ans=max(ans,l1+l2+link[x]-1);
}
int main()
{
n=read();m=read();
for(int i=1;i<=m;++i)
{
int x,y;x=read();y=read();
add(x,y);add(y,x);
link[x]++;link[y]++;
}
for(int i=1;i<=n;++i) f[i]=1;
dfs(1,0);
printf("%d",ans);
return 0;
}
勿忘初心！
