【题】【树(树的直径&LCA)】NKOJ 2044 奶牛政坛 【RMQ&LCA】

NKOJ 2044 奶牛政坛 【RMQ&LCA】
时间限制 : 10000 MS 空间限制 : 65536 KB

问题描述
农夫约翰的奶牛住在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之间的距离）。帮助奶牛们求出每个政党的范围。

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

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

样例输入
6 2
1 3
2 1
1 0
2 1
2 1
1 5

样例输出
3
2

来源 Usaco2010 Hol

思路：
对于每个政党找一次只有关于该政党的点的树的直径即为答案
具体实现
枚举每个点，如果没有讨论过它所属的政党，就把改点选出来，以后每遇到一个属于该政党的点就与选出来的的点求距离，不断更新temp值。
第二遍，用temp中存储的点找同一政党最远值，即为答案

代码：

#include<cstdio>
#include<iostream>
#include<queue> 
#include<cmath>
#include<cstdio>
using namespace std;
const int need=200003;
const int needk=100004;
const double log_2=log(2);

int head[needk],belong[needk],temp[needk],now_temp[needk],ans[needk];
int dep[need],fa[need][19],s=0;
int en[need*2],la[need*2],fi[need],m=0;

void add(int a,int b)
{
    m++;
    la[m]=fi[a];
    fi[a]=m;
    en[m]=b;
} 

void dfs(int x)
{
    int k=ceil(log(dep[x])/log_2);
    s=max(s,k);
    for(int i=1;i<=s;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
    int t=fi[x],y;
    while(t)
    {
        y=en[t];
        fa[y][0]=x;
        dep[y]=dep[x]+1;
        dfs(y);
        t=la[t]; 
    } 
}

void go_up(int &u,int &v)
{
    if(dep[u]>dep[v]) swap(u,v);
    int p=dep[v]-dep[u];
    for(int i=0;i<=s;i++) if(p&(1<<i)) v=fa[v][i];
}

int lca(int u,int v)
{
    go_up(u,v);
    if(u==v) return v;
    for(int i=s;i>=0;i--) if(fa[v][i]!=fa[u][i]) v=fa[v][i],u=fa[u][i];
    return fa[v][0]; 
}

int cnt(int u,int v)
{
    int c=lca(u,v);
    return dep[u]+dep[v]-(dep[c]<<1);
}


int main()
{
    int n,k;scanf("%d%d",&n,&k);
    int x;
    for(int i=1,a,b;i<=n;i++)
    {
        scanf("%d%d",&belong[i],&a);
        if(a==0)x=i;
        else add(a,i);
    }
    dfs(x);
    for(int i=1,a;i<=n;i++)
    {
        if(head[belong[i]]==0) head[belong[i]]=i;
        else  
        {
            a=cnt(head[belong[i]],i);
            if(a>now_temp[belong[i]])
            {
                temp[belong[i]]=i;
                now_temp[belong[i]]=a;
            }
        }
    }
    for(int i=1,a;i<=n;i++)
    {
        a=cnt(temp[belong[i]],i);
        if(a>ans[belong[i]]) ans[belong[i]]=a;
    }
    for(int i=1;i<=k;i++)printf("%d\n",ans[i]);
}