无聊的草稿

Description
　图中有N个点，每两点间只有唯一的路径，对于这样一个给定的图，最大的“毛毛虫”会有多大。毛毛虫包含一条主链，毛毛虫中的节点，要不在主链上，要么和主链上某节点相邻，如下图所示有两只合法的毛毛虫，点数越多，毛毛虫越大。

Input
　　输入文件第一行两个整数N，M（N≤1000000）
　　接下来M行，每行两个整数a, b(a, b≤N)
　　你可以假定没有一对相同的(a, b)会出现一次以上。
Output
　　一个整数ans,表示最大的毛毛虫的大小。
Sample Input
5 4
1 2
1 3
4 1
5 1

The Solution

这道题的题目大意就是要求一段树上最长链。。。

我们可以先将每个点转化为一个带权的点，权为该点的度数，
然后就和在一棵树上求最长链一样了。。。

可以用两个dfs求。

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define N 1000005
using namespace std;
int t[N*2],last[N*2],next[N*2],a[N*2];
int n,m,l=0,q=0,ans=0;
void add(int x,int y)
{
    a[x]++;
    t[++l]=y;
    next[l]=last[x];
    last[x]=l;
}
void dg(int x,int y,int z)
{
    if (z>ans) ans=z,q=x;       
    for (int k=last[x];k;k=next[k])
        if (t[k]!=y) dg(t[k],x,z+a[t[k]]-1);    
}
int main()
{
    int x,y;
    scanf("%d%d",&n,&m);
    fo(i,1,m) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dg(1,0,a[1]+1);
    dg(q,0,a[q]+1);
    printf("%d\n",ans);
}