P3174 [HAOI2009]毛毛虫
题目描述
对于一棵树,我们可以将某条链和与该链相连的边抽出来,看上去就象成一个毛毛虫,点数越多,毛毛虫就越大。例如下图左边的树(图 1 )抽出一部分就变成了右边的一个毛毛虫了(图 2 )。
输入输出格式
输入格式:在文本文件 worm.in 中第一行两个整数 N , M ,分别表示树中结点个数和树的边数。
接下来 M 行,每行两个整数 a, b 表示点 a 和点 b 有边连接( a, b ≤ N )。你可以假定没有一对相同的 (a, b) 会出现一次以上。
输出格式:在文本文件 worm.out 中写入一个整数 , 表示最大的毛毛虫的大小。
输入输出样例
说明
40% 的数据, N ≤ 50000
100% 的数据, N ≤ 300000
#include<bits/stdc++.h> using namespace std; const int maxn = 3e5+10; int n,m,size,f[maxn],head[maxn],sum[maxn],ans; struct edge { int v,nex; }e[maxn<<1]; void adde(int u,int v) { e[size].v=v;e[size].nex=head[u];head[u]=size++; } void dfs(int u,int fa) { sum[u]=1; for(int i=head[u];~i;i=e[i].nex) { int v=e[i].v; if(v==fa) continue; sum[u]++; } f[u]=sum[u]; for(int i=head[u];~i;i=e[i].nex) { int v=e[i].v; if(v==fa) continue; dfs(v,u); ans=max(ans,f[u]+f[v]-1+(u!=1)); f[u]=max(f[u],f[v]+sum[u]-1); } } int main() { memset(head,-1,sizeof(head)); scanf("%d%d",&n,&m); for(int i=1;i<=m;i++) { int u,v;scanf("%d%d",&u,&v); adde(u,v);adde(v,u); } dfs(1,-1); printf("%d",ans); return 0; }