Jzoj3555 树的直径

来源:互联网 发布:淘宝退款不退货会怎样 编辑:程序博客网 时间:2024/06/05 10:43
科学家在观测一棵大树,这棵树在不断地生长,科学家给这棵树的每个节点编了号。开始的时候,这棵树很小只有4个节点,一号点为根,其他三个节点挂在上面。

在接下来的M次观察中,科学家每次都能看见这棵树从叶子处长出新的两个节点来。如果当前这棵树有N个节点,那么这棵树的新的两个节点的编号分别为N+1,N+2。科学家记录下了这棵树生长的过程,需要你帮着计算这棵树实时的直径。树的直径就是这棵树最远的两个节点的距离。

每行一个整数x,代表这棵树的编号为x的节点下面又长了两个叶子节点。保证每次生长的节点都是叶子节点

好久没在jzoj上切题了,估计ranking都要掉了赶紧做一做今天的模拟题

还好第一题还比较简单,直接将新建节点和原来直径合并就好了,求lca就用倍增ok

#pragma GCC opitmize("O3")#pragma G++ opitmize("O3")#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;int f[200010][18],d[200010],n;inline void link(int p,int x){f[x][0]=p; d[x]=d[p]+1;for(int i=1;i<18;++i) f[x][i]=f[f[x][i-1]][i-1];}inline void swim(int& x,int d){for(int i=0;d;++i,d>>=1) if(d&1) x=f[x][i];}inline int gLca(int x,int y){if(d[x]>d[y]) swap(x,y);swim(y,d[y]-d[x]);if(x==y) return x;for(int j=17;;){for(;~j&&f[x][j]==f[y][j];--j);if(j<0) return f[x][0];x=f[x][j]; y=f[y][j];}}int main(){d[2]=d[3]=d[4]=1; *f[1]=0;f[2][0]=f[3][0]=f[4][0]=1;scanf("%d",&n); int x=2,y=3,r=2;for(intp,c,t=4;n--;){scanf("%d",&c);link(c,++t); link(c,++t);p=gLca(x,t);if(d[x]+d[t]-2*d[p]>r){ r=d[x]+d[t]-2*d[p]; y=t; }p=gLca(y,t);if(d[y]+d[t]-2*d[p]>r){ r=d[y]+d[t]-2*d[p]; x=t; }printf("%d\n",r);}}