CodeForces - 379F - New Year Tree（树的直径）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_40032278/article/details/81556776

题目

原始有一棵根为１，有三个叶子２，３，４的树。有ｔ个操作，每次可以在一个叶子下面续上两个节点，每次操作完问当前树的直径。

题解

1.每次加点后，树的直径的端点至多只会变化一个，且必然有一个端点是新加的点。
2. 所以只要分别求他到原直径两端的距离，若大于原直径长度则更新直径。

#include<bits/stdc++.h>
using namespace std;
int a[1000010][30],d[1000010],n,s,t,mmax;
void init()
{
    memset(a,0,sizeof(a));
    d[1]=0;
    d[2]=d[3]=d[4]=1;
    a[2][0]=a[3][0]=a[4][0]=1;
    n=4;
    s=2;
    t=4;
    mmax=2;
}
void build(int u, int fa)
{
    a[u][0]=fa;
    d[u]=d[fa]+1;
    for(int i=1; i<=21; i++)
        a[u][i]=a[a[u][i-1]][i-1];
}

int sol(int x, int y)
{
    if(x==y) return 0;
    if(d[x]<d[y]) swap(x,y);
    int len=0,i;
    for(i=21; i>=0; i--)
        if(d[x]-(1<<i)>=d[y])
            len+=(1<<i),x=a[x][i];
    for(i=21; i>=0; i--)
        if(a[x][i]!=a[y][i])
            len+=2*(1<<i),x=a[x][i],y=a[y][i];

    if(x!=y) len+=2;
    return len;
}
int main()
{
    int q,x,l;
    init();
    scanf("%d",&q);
    while(q--)
    {
        scanf("%d",&x);
        build(n+1,x);
        build(n+2,x);
        l=sol(n+1,s);
        if(l>mmax)
        {
            t=n+1;
            mmax=l;
        }
        l=sol(n+1,t);
        if(l>mmax)
        {
            s=n+1;
            mmax=l;
        }
        printf("%d\n", mmax);
        n=n+2;
    }
    return 0;
}