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 
#include 
#include 
#include 
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; id[b]) swap(a,b);
    if (d[a]=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+2l2)
        {
            cout<



你可能感兴趣的:(LCA,codeforces)