当时做这套题的时候不会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; }