JZOJ 1752. 无聊的草稿

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

Sample Output

5

Data Constraint

Hint

【数据规模】
　　1. 对于20%的数据，N≤200
　　2. 对于40%的数据，N≤5000
　　3. 对于100%的数据，N≤10^6

分析

其实就是找树的直径。

代码

#include <algorithm>
#include <iostream>
#include <cstring>
#include <complex>
#include <cstdio>
#include <queue>
#include <cmath>
#include <map>
#include <set>

#define N 1000005
#define INF 0x7fffffff
#define sqr(x) ((x) * (x))
#define pi acos(-1)

int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9'){if (ch == '-') f = -1; ch = getchar();}
    while (ch >= '0' && ch <= '9')  {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

struct NOTE
{
    int to,next;
}e[N * 2];

int next[N];

int cnt;

int f[N];

int d[N];

void add(int x,int y)
{
    e[++cnt] = (NOTE){y, next[x]}; next[x] = cnt;
    e[++cnt] = (NOTE){x, next[y]}; next[y] = cnt;
}

int n,m;

bool vis[N];

int ans;

int x;

void dfs(int p, int sum, int dep)
{
    vis[p] = false;
    int s;
    if (dep == 1)
        s = d[p];
    else s = d[p] - 1;
    for (int i = next[p]; i; i = e[i].next)
    {
        if (vis[e[i].to])
            dfs(e[i].to, sum + s, 0);
    }
    if (sum > ans)
        ans = sum, x = p;
}

int main()
{
    n = read(); m = read();
    for (int i = 1; i <= m; i++)
    {
        int x = read(), y = read();
        add(x,y);
        d[x]++; d[y]++;
    }
    memset(vis, 1, sizeof(vis));
    ans = 0;
    dfs(1,1,1);
    memset(vis, 1, sizeof(vis));
    ans = 0;
    dfs(x,1,1);
    printf("%d\n",ans);
}