COGS——T 803. [USACO Hol10] 政党 || 1776: [Usaco2010 Hol]cowpol 奶牛政坛

http://www.lydsy.com/JudgeOnline/problem.php?id=1776||http://cogs.pro/cogs/problem/problem.php?pid=803

Time Limit: 10 Sec Memory Limit: 64 MB
Submit: 507 Solved: 246
[Submit][Status][Discuss]

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

HINT

Source

Gold

距离最远的两个点之中，一定有一个是在当前政党中深度最深的，枚举另一个点，更新政党范围

mdzz我居然用点编号和深度去比较！！

 1 #include <cstdio>
 2 
 3 const int N(2e5+5);
 4 int a[N],p[N],deps[N],ans[N];
 5 int head[N],sumedge;
 6 struct Edge
 7 {
 8     int v,next;
 9     Edge(int v=0,int next=0):v(v),next(next){}
10 }edge[N<<1];
11 inline void ins(int u,int v)
12 {
13     edge[++sumedge]=Edge(v,head[u]);
14     head[u]=sumedge;
15     edge[++sumedge]=Edge(u,head[v]);
16     head[v]=sumedge;
17 }
18 
19 #define max(a,b) (a>b?a:b)
20 #define swap(a,b) {int tmp=a;a=b,b=tmp;}
21 int dad[N],dep[N],size[N],son[N],top[N];
22 void DFS(int u)
23 {
24     size[u]=1;
25     dep[u]=dep[dad[u]]+1;
26     for(int v,i=head[u];i;i=edge[i].next)
27     {
28         v=edge[i].v;
29         if(dad[u]==v) continue;
30         dad[v]=u; DFS(v); size[u]+=size[v];
31         if(size[son[u]]<size[v]) son[u]=v;
32     }
33 }
34 void DFS_(int u,int Top)
35 {
36     top[u]=Top;
37     if(son[u]) DFS_(son[u],Top);
38     for(int v,i=head[u];i;i=edge[i].next)
39     {
40         v=edge[i].v;
41         if(dad[u]!=v&&son[u]!=v) DFS_(v,v);
42     }
43 }
44 int LCA(int x,int y)
45 {
46     for(;top[x]!=top[y];y=dad[top[y]])
47         if(dep[top[x]]>dep[top[y]]) swap(x,y);
48     return dep[x]<dep[y]?x:y;
49 }
50 
51 inline void read(int &x)
52 {
53     x=0; register char ch=getchar();
54     for(;ch>'9'||ch<'0';) ch=getchar();
55     for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
56 }
57 inline void write(int x)
58 {
59     if(x/10) write(x/10);
60     putchar(x%10+'0');
61 }
62 int AC()
63 {
64 //    freopen("cowpol.in","r",stdin);
65 //    freopen("cowpol.out","w",stdout);
66     
67     int n,k,rt; read(n),read(k);
68     for(int i=1;i<=n;i++)
69     {
70         read(a[i]),read(p[i]);
71         if(!p[i]) rt=i;
72         else ins(p[i],i);
73     }
74     DFS(rt); DFS_(rt,rt);
75     for(int i=1;i<=n;i++)
76         if(dep[i]>dep[deps[a[i]]]) deps[a[i]]=i;
77     for(int lca,i=1;i<=n;i++)
78     {
79         lca=LCA(i,deps[a[i]]);
80         ans[a[i]]=max(ans[a[i]],dep[i]+dep[deps[a[i]]]-dep[lca]*2);
81     }
82     for(int i=1;i<=k;i++)
83         write(ans[i]),puts("");
84     return 0;
85 }
86 
87 int I_want_AC=AC();
88 int main(){;}