题目描述

Description

农夫约翰的奶牛住在N (2 <= N <= 200,000)片不同的草地上，标号为1到N。恰好有N-1条单位长度的双向道路，用各种各样的方法连接这些草地。而且从每片草地出发都可以抵达其他所有草地。也就是说，这些草地和道路构成了一种叫做树的图。输入包含一个详细的草地的集合，详细说明了每个草地的父节点P_i (0 <= P_i <= N)。根节点的P_i == 0, 表示它没有父节点。因为奶牛建立了1到K一共K (1 <= K <= N/2)个政党。每只奶牛都要加入某一个政党，其中， 第i只奶牛属于第A_i (1 <= A_i <= K)个政党。而且每个政党至少有两只奶牛。 这些政党互相吵闹争。每个政党都想知道自己的“范围”有多大。其中，定义一个政党的范围是这个政党离得最远的两只奶牛（沿着双向道路行走）的距离。 比如说，记为政党1包含奶牛1，3和6，政党2包含奶牛2，4和5。这些草地的连接方式如下图所 示（政党1由-n-表示）： 政党1最大的两只奶牛的距离是3（也就是奶牛3和奶牛6的距离）。政党2最大的两只奶牛的距离是2（也就是奶牛2和4，4和5，还有5和2之间的距离）。 帮助奶牛们求出每个政党的范围。

Input

• 第一行: 两个由空格隔开的整数: N 和 K * 第2到第N+1行: 第i+1行包含两个由空格隔开的整数: A_i和P_i

Output

• 第1到第K行: 第i行包含一个单独的整数，表示第i个政党的范围。

Sample Input

6 2
1 3
2 1
1 0
2 1
2 1
1 5

Sample Output

3
2

题目分析

贪心的想 每个政党里,离得最远的两个奶牛中 肯定有一个是深度最大的
所以求出深度最大的点后更新答案
使用vector+lca 树剖lca还不错

#include <cstdio>
#include <cstring>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;
int n;
vector<int>a[100100];
int head[200100],to[400100],net[400100],tot;
void add(int x,int y)
{
    net[++tot]=head[x],head[x]=tot,to[tot]=y;
}
int deep[200100],size[200100],fa[200100];
int top[200100],son[200100];
void dfs(int x)
{
    size[x]=1,deep[x]=deep[fa[x]]+1;
    for(int i=head[x];i;i=net[i])
    {
        dfs(to[i]),size[x]+=size[to[i]];
        if(size[son[x]]<size[to[i]]) son[x]=to[i];
    }
}
void dfs2(int x,int temp)
{
    top[x]=temp;
    if(son[x]) dfs2(son[x],temp);
    for(int i=head[x];i;i=net[i])
        if(to[i]!=son[x]&&to[i]!=fa[x]) dfs2(to[i],to[i]);
}
int lca(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    if(deep[x]>deep[y]) swap(x,y);
    return x;
}
int k,root;
int main()
{   
    scanf("%d%d",&n,&k);
    for(int x,i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&fa[i]);
        if(fa[i]) add(fa[i],i);
        else root=i;
        a[x].push_back(i);
    }
    dfs(root),dfs2(root,root);
    for(int i=1;i<=k;i++)
    {
        if(!a[i].size())
        {
            printf("0\n");
            continue;
        }
        int tmp=0;
        for(int j=0;j<a[i].size();j++)
            if(deep[a[i][tmp]]<deep[a[i][j]]) tmp=j;
        int ans=0;
        for(int j=0;j<a[i].size();j++)
            if(j!=tmp) 
                ans=max(ans,deep[a[i][j]]+deep[a[i][tmp]]-2*deep[lca(a[i][j],a[i][tmp])]);
        printf("%d\n",ans);
    }
    return 0;
}