bzoj:1776: [Usaco2010 Hol]cowpol 奶牛政坛

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

 

怎么你们都会结论……

怎么你们都写lca……

怎么就我写点分啊……

怎么都是nlogn，你们跑那么快，就我跑6s啊……

为什么……为什么事情会变成这样子呢！？

 

/**************************************************************
    Problem: 1776
    User: JSZX11556
    Language: C++
    Result: Accepted
    Time:6300 ms
    Memory:17468 kb
****************************************************************/
 
#include<queue>
#include<cstdio>
#include<algorithm>
#define MN 200001
using namespace std;
 
int read_p,read_ca,read_f;
inline int read(){
    read_p=0;read_ca=getchar();read_f=1;
    while(read_ca<'0'||read_ca>'9') read_f=read_ca=='-'?-1:read_f,read_ca=getchar();
    while(read_ca>='0'&&read_ca<='9') read_p=read_p*10+read_ca-48,read_ca=getchar();
    return read_p*read_f;
}
struct na{int y,ne;}b[MN<<1];
int n,m,f,p[MN],ro,l[MN],num=0,s[MN],mi,S,mmh[MN],MMH[MN];
bool bo[MN];
int q[MN],top=0;
inline void in(int x,int y){if (x&&y)b[++num].y=y,b[num].ne=l[x],l[x]=num;}
void gr(int x,int f){
    s[x]=1;
    int o=0;
    for (int i=l[x];i;i=b[i].ne)
    if (b[i].y!=f&&!bo[b[i].y]){
        gr(b[i].y,x);
        if (o<s[b[i].y]) o=s[b[i].y];
        s[x]+=s[b[i].y];
    }
    if (S-s[x]>o) o=S-s[x];
    if (o<mi) mi=o,ro=x;
}
void dfs(int x,int f,int d){
    s[x]=1;
    if (mmh[p[x]]+d>MMH[p[x]]&&mmh[p[x]]!=-1) MMH[p[x]]=mmh[p[x]]+d;
    if (p[x]==p[ro]&&d>MMH[p[x]]) MMH[p[x]]=d;
    for (int i=l[x];i;i=b[i].ne) if (b[i].y!=f&&!bo[b[i].y]) dfs(b[i].y,x,d+1),s[x]+=s[b[i].y];
}
void DFS(int x,int f,int d){
    if (mmh[p[x]]==-1) q[++top]=p[x];
    if (d>mmh[p[x]]) mmh[p[x]]=d;
    for (int i=l[x];i;i=b[i].ne) if (b[i].y!=f&&!bo[b[i].y]) DFS(b[i].y,x,d+1);
}
void work(int x){
    mi=1e9;gr(x,0);x=ro;bo[x]=1;
    for (int i=l[x];i;i=b[i].ne)
    if (!bo[b[i].y]) dfs(b[i].y,x,1),DFS(b[i].y,x,1);
    while (top) mmh[q[top--]]=-1;
    for (int i=l[x];i;i=b[i].ne)
    if (!bo[b[i].y]) S=s[b[i].y],work(b[i].y);
}
int main(){
    register int i;
    n=read();m=read();
    for (i=1;i<=n;i++) p[i]=read(),f=read(),in(f,i),in(i,f);
    for (i=1;i<=m;i++) mmh[i]=-1;
    S=n;work(1);
    for (i=1;i<=m;i++) printf("%d\n",MMH[i]);
}