一棵树,根节点为1,有2,3,4三个叶子节点,接下来有N个操作,每个操作x,表示给节点x添加两个孩子节点,并且输出添加后树的直径。
当时做这套题的时候不会LCA,写了个暴力的爬山坡,结果复测时喜闻乐见的TLE了= =...现在学了LCA要果断虐回来...分别每次记录直径的两个端点各自的父亲节点d1,d2,初始值d1=d2=1,然后每次添加时,找一下x到d1和x到d2的距离那个更大,如果两个距离都小于上一次的答案的话,d1,d2不变,输出上一次的直径,否则用x去更新距离较小的d得到新的d1,d2,而此时的直径也就是新的d1,d2的距离+2(因为d1,d2是直径两端点各自的父亲节点,所以一边加一个1)...貌似有更简单的方法...我这个CF上跑了1700ms,看到有人一个while循环搞下来才几百毫秒...回头去研究下..
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=1250000;
const int POW=20;
int n,m;
struct EDGE
{
int v,w,next;
}edge[maxn<<1];
int g[maxn];
int en,q,k;
int fa[maxn];
void addedge(int x,int y)
{
fa[y]=x;
edge[en].v=y;
edge[en].next=g[x];
g[x]=en;
en++;
edge[en].v=x;
edge[en].next=g[y];
g[y]=en;
en++;
}
int d[maxn],p[maxn][26];
int dis[maxn];
void dfs(int u,int fa)
{
d[u]=d[fa]+1;
p[u][0]=fa;
for (int i=1; i<POW; i++) p[u][i]=p[p[u][i-1]][i-1];
for (int j=g[u]; j!=-1; j=edge[j].next)
{
int v=edge[j].v;
if (v==fa) continue;
dfs(v,u);
dis[v]=dis[u]+1;
}
}
int lca(int a,int b)
{
if (d[a]>d[b]) swap(a,b);
if (d[a]<d[b])
{
int del=d[b]-d[a];
for (int i=0; i<POW; i++)if (del&(1<<i)) b=p[b][i];
}
if (a!=b)
{
for (int i=POW-1; i>=0; i--)
{
if (p[a][i]!=p[b][i])
a=p[a][i],b=p[b][i];
}
a=p[a][0];
b=p[b][0];
}
return a;
}
int main()
{
// freopen("in.txt","r",stdin);
scanf("%d",&m);
en=0;
memset(g,-1,sizeof g);
memset(fa,0,sizeof fa);
addedge(1,2);
addedge(1,3);
addedge(1,4);
int x;
int cnt=4;
dfs(1,-1);
int d1=1,d2=1;
int last=1;
for (int i=1; i<=m; i++)
{
scanf("%d",&x);
cnt++;
addedge(x,cnt);
cnt++;
addedge(x,cnt);
dfs(x,fa[x]);
int t1,t2,l1,l2;
t1=lca(x,d1);
t2=lca(x,d2);
l1=dis[x]+dis[d1]-2*dis[t1];
l2=dis[x]+dis[d2]-2*dis[t2];
if (l1+2<last && l2+2<last)
{
cout<<last<<endl;
}
else
if (l1>l2)
{
cout<<l1+2<<endl;
last=l1+2;
d2=x;
}
else
{
cout<<l2+2<<endl;
last=l2+2;
d1=x;
}
}
return 0;
}