P3174 [HAOI2009]毛毛虫(树形dp)

P3174 [HAOI2009]毛毛虫

题目描述

对于一棵树，我们可以将某条链和与该链相连的边抽出来，看上去就象成一个毛毛虫，点数越多，毛毛虫就越大。例如下图左边的树（图 1 ）抽出一部分就变成了右边的一个毛毛虫了（图 2 ）。

输入输出格式

输入格式：

在文本文件 worm.in 中第一行两个整数 N ， M ，分别表示树中结点个数和树的边数。

接下来 M 行，每行两个整数 a, b 表示点 a 和点 b 有边连接（ a, b ≤ N ）。你可以假定没有一对相同的 (a, b) 会出现一次以上。

输出格式：

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

输入输出样例

输入样例#1：复制

输出样例#1：复制

说明

40% 的数据， N ≤ 50000

100% 的数据， N ≤ 300000

/*
类似dfs求直径
r[]为入度
一条毛毛虫的点数为 
Σr[i]-(链长-1)+1 
=Σ(r[i]-1)+2 
*/ 
#include<iostream>
#include<cstdio>
#include<cstring>#define N 300007using namespace std;
int n,m,ans,cnt,S,T;
int head[N],deep[N],pre[N];
int r[N];
struct edge{int u,v,net;
}e[N<<1];inline int read()
{int x=0,f=1;char c=getchar();while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}return x*f;
}inline void add(int u,int v)
{e[++cnt].v=v;e[cnt].net=head[u];head[u]=cnt;
}void dfs(int u,int fa,int num)
{if(num>ans) S=u,ans=num; for(int i=head[u];i;i=e[i].net){int v=e[i].v;if(v==fa) continue;dfs(v,u,num+r[v]);}
}int main()
{int x,y;n=read();m=read();memset(r,-1,sizeof r);for(int i=1;i<n;i++){x=read();y=read();add(x,y);add(y,x);r[x]++;r[y]++;}dfs(1,0,r[1]);ans=0;dfs(S,0,r[S]);printf("%d\n",ans+2);return 0; 
}

/*
把链连带链上的点的儿子算作一条链 
f[x]表示以x为根最大链的大小
维护以某个点为根最大链和次大链计算答案 
*/
#include<iostream>
#include<cstdio>
#include<cstring>#define N 300007using namespace std;
int f[N],head[N],son[N];
int n,m,k,ans,maxx;
struct edge
{int to,net;
}e[N<<1];inline void add(int u,int v)
{e[++k].to=v;e[k].net=head[u];head[u]=k;
}inline int read()
{int x=0,f=1;char c=getchar();while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}return x*f;
}void dfs(int u,int fa)
{int v,bigx=0,lowx=0;for(int i=head[u];i;i=e[i].net){v=e[i].to;if(v!=fa){dfs(v,u);if(f[v]>lowx)//维护最大链与次大链 
            {if(f[v]>bigx)lowx=bigx,bigx=f[v];else lowx=f[v];}f[u]=max(f[u],f[v]+son[u]-1);}}ans=max(ans,lowx+bigx+son[u]-1);//(-1是因为根节点重复加了) 
}int main()
{n=read();m=read();for(int i=1;i<=m;i++){int u,v;u=read();v=read();add(u,v);add(v,u);son[u]++;son[v]++; }for(int i=1;i<=n;i++)f[i]=1;dfs(1,0);printf("%d",ans);
}