[haoi2009]毛毛虫 树形dp

时间:2023-03-09 08:17:28
[haoi2009]毛毛虫 树形dp

这道题细节处理不少,但要AC不难;

设以i节点为根节点的子树能形成的最大的毛毛虫长度为f[i],则f[i]=max(f[j])+i节点的孩子数;

答案需要f最大和次大的两个子树合并,而且若合并的位置不是根节点,ans++;

我就是坑在了最后一点上,最后打表找到了问题;

 #include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=;
int n,m,f[maxn],child[maxn];
struct node{
int y,next;
}e[maxn<<];
int linkk[maxn],len=;
void insert(int x,int y){
e[++len].y=y;
e[len].next=linkk[x];
linkk[x]=len;
}
void init(){
scanf("%d%d",&n,&m);
int x,y;
for(int i=;i<=m;i++){
scanf("%d%d",&x,&y);
insert(x,y);insert(y,x);
}
}
void dfs(int x,int fa){
for(int i=linkk[x];i;i=e[i].next){
if(e[i].y==fa)continue;
child[x]++;
dfs(e[i].y,x);
}
}
int ans=;
void Dfs(int x,int fa){
if(child[x]==){
f[x]=;return;
}
int maxx[]={,};
for(int i=linkk[x];i;i=e[i].next){
if(e[i].y==fa)continue;
Dfs(e[i].y,x);
f[x]=max(f[x],f[e[i].y]+child[x]);
if(f[e[i].y]>maxx[])maxx[]=f[e[i].y];
if(f[e[i].y]>maxx[])maxx[]=maxx[],maxx[]=f[e[i].y];
}
ans=max(ans,f[x]+maxx[]-);
if(x!=)ans=max(ans,f[x]+maxx[]);
if(maxx[]==)ans=max(ans,f[x]);
}
void work(){
dfs(,);
Dfs(,);
printf("%d\n",ans);
}
int main(){
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
init();
work();
}