codeforces 379F New Year Tree 在线LCA

       一棵树,根节点为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;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值