題目描述

對於一棵樹，我們可以將某條鏈和與該鏈相連的邊抽出來，看上去就象成一個毛毛蟲，點數越多，毛毛蟲就越大。例如下圖左邊的樹（圖 1 ）抽出一部分就變成了右邊的一個毛毛蟲了（圖 2 ）。

輸入輸出格式

輸入格式：

在文本文件 worm.in 中第一行兩個整數 N ， M ，分別表示樹中結點個數和樹的邊數。

接下來 M 行，每行兩個整數 a, b 表示點 a 和點 b 有邊連接（ a, b ≤ N ）。你可以假定沒有一對相同的 (a, b) 會出現一次以上。

輸出格式：

在文本文件 worm.out 中寫入一個整數 , 表示最大的毛毛蟲的大小。

輸入輸出樣例

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 

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;
}

勿忘初心！

[HAOI2009]毛毛蟲