【BZOJ3910】火车 LCA+并查集

最新推荐文章于 2020-02-23 22:18:57 发布

原创最新推荐文章于 2020-02-23 22:18:57 发布 · 1.6k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#BZOJ3910 #火车 #LCA #并查集

lca 同时被 3 个专栏收录

10 篇文章

订阅专栏

并查集

7 篇文章

订阅专栏

并查集的高级应用

4 篇文章

订阅专栏

该博客介绍了如何利用倍增LCA和并查集解决BZOJ3910题目的火车问题。通过倍增LCA算法找到两点之间的最短路径，并使用并查集进行路径标记，实现线性时间复杂度的解决方案。

链接：

#include <stdio.h>
int main()
{
    puts("转载请注明出处[vmurder]谢谢");
    puts("网址：blog.csdn.net/vmurder/article/details/44780959");
}

题解：

首先找两点之间路径可以用倍增LCA。
然后标记哪个点走过可以用并查集，均摊下来最后是线性的。

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 501000
#define LOGN 20
using namespace std;
struct Eli
{
    int v,next;
}e[N<<1];
int head[N],cnt;
inline void add(int u,int v)
{
    e[++cnt].v=v;
    e[cnt].next=head[u];
    head[u]=cnt;
}
int n,m,root;
int dep[N],fa[N][LOGN];
void dfs(int x,int p)
{
    int i,v;
    dep[x]=dep[p]+1;
    for(i=head[x];i;i=e[i].next)
    {
        v=e[i].v;
        if(v==p)continue;
        fa[v][0]=x;
        dfs(v,x);
    }
}
int f[N];int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
void array()
{
    for(int i=1;i<=n;i++)f[i]=i;
    for(int j=1;j<LOGN;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
}
int getlca(int &ans,int x,int y)
{
    ans=0;
    int i,j;
    if(dep[x]<dep[y])swap(x,y);
    for(i=LOGN-1;~i;i--)
        if(dep[fa[x][i]]>=dep[y])
            ans+=1<<i,x=fa[x][i];
    if(x==y)return x;
    for(i=LOGN-1;~i;i--)
        if(fa[x][i]!=fa[y][i])
            ans+=1<<i+1,x=fa[x][i],y=fa[y][i];
    ans++,ans++;
    return fa[x][0];
}
int main()
{
//  freopen("test.in","r",stdin);
    int i,j,k;
    int a,b,c;

    scanf("%d%d%d",&n,&m,&root);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    dfs(1,0),array();
    long long ans=0;
    for(int pos=root;m--;)
    {
        scanf("%d",&a);
        if(find(a)!=a)continue;

        int lca=getlca(c,a,pos);
        ans+=c;

        int x=find(a),y=find(pos);
        while(x!=y)
        {
            if(dep[x]<dep[y])swap(x,y);
            x=f[x]=find(fa[x][0]);
        }
        if(find(lca)==lca)f[lca]=fa[lca][0];
        pos=a;
    }
    cout<<ans<<endl;
}