Jzoj3555 树的直径

科学家在观测一棵大树,这棵树在不断地生长,科学家给这棵树的每个节点编了号。开始的时候,这棵树很小只有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(int	p,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);
	}
}

posted @ 2017-12-02 21:19  扩展的灰(Extended_Ash)  阅读(162)  评论(0编辑  收藏  举报