首页 > 其他分享 >boruvka

boruvka

时间:2024-12-25 09:13:19浏览次数:2  
标签:int s2 s1 fa boruvka sum se

boruvka

是一种对于完全图求最小生成树很好用的算法。

算法流程

每轮为当前每个连通块找到与其最近的连通块,并连边,直到只有一个连通块。

正确性

最后的最小生成树上的每个点,显然都会保留它连出的最短的边。

否则断掉现在它连出的一条边,再连最短的边一定更优。

那么每轮过后,把一个连通块缩成一个点,按照上面的结论一直做下去就是对的。

轮数

每次连通块个数至少除以二,因为最坏情况下的连边就是1-2,3-4,5-6,... \(n-1\) -\(n\)。

所以最多有 \(\log n\) 轮。

例题

Atcoder_cf17_final_j Tree MST

做法挺巧妙的。

我们先算出每个点和不同连通块的点的最短边长,以及最近点在哪个块,然后合并到这个点的连通块上去。

设 \(sum_i\) 表示点 \(i\) 到根节点的树上距离,\(a_i\) 表示点 \(i\) 的权值。

考虑树形 \(dp\),先 \(dfs\) 一遍找出子树内最近的点(其实就是子树里 \(sum_v+a_v\) 最小的点,因为 \(u\) 到 \(v\) 的边长是 \(a_u+a_v+sum_v-sum_u\))。

但是!这个点可能是在同一块里的,怎么办呢?

我们可以不只维护最近的点,另外再维护一个和当前最近点不是同一块的次近点。

这样的话,如果最近点在同一块里,次近点就是答案了。

第二遍 \(dfs\),我们求出子树外最近的点,子树外的点到 \(u\) 的边长是 \(a_u+a_v+sum_v+sum_u-2\times sum_{lca}\)。当 \(lca\) 固定的时候,这个 \(v\) 一定是上一次 \(dfs\) 里算出的 \(lca\) 的子树中, \(sum_v+a_v\) 最小的点、和最小点不在一个块的次小点。那么维护从根节点到 \(u\),\(a_v+sum_v-2\times sum_{lca}\) 最小的点,和不在同一块的次小点就好了。

代码:

#include<bits/stdc++.h>
#define int long long
#define mkp make_pair
#define fi first
#define se second
using namespace std;
const int N=2e5+10;
int n,cnt,res,a[N],f[N],s[N];
int idx,hd[N],to[N<<1],nxt[N<<1],len[N<<1];
pair<int,int>p,s1[N],s2[N],ans[N];
int find(int x)
{
    if(f[x]==x)return x;
    return f[x]=find(f[x]);
}
void add(int u,int v,int w)
{
    ++idx,to[idx]=v,nxt[idx]=hd[u],len[idx]=w,hd[u]=idx;
    return;
}
pair<int,int>cmx(pair<int,int>a,pair<int,int>b,pair<int,int>c,int f)
{
    p=mkp(1e18,-1);
    if(a.se!=f)p=min(p,a);
    if(b.se!=f)p=min(p,b);
    if(c.se!=f)p=min(p,c);
    return p;
}
void dfs1(int u,int fa)
{
    s1[u]=mkp(s[u]+a[u],f[u]);
    s2[u]=mkp(1e18,-1);
    for(int i=hd[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa)continue;
        s[v]=s[u]+len[i];
        dfs1(v,u);
        if(s1[u]>s1[v])s2[u]=cmx(s1[u],s2[u],s2[v],s1[v].se),s1[u]=s1[v];
        else s2[u]=cmx(s2[u],s1[v],s2[v],s1[u].se);
    }
    return;
}
void dfs2(int u,int fa)
{
    s1[u].fi-=2*s[u],s2[u].fi-=2*s[u];
    if(fa)if(s1[u]>s1[fa])s2[u]=cmx(s2[fa],s1[u],s2[u],s1[fa].se),s1[u]=s1[fa];
    else s2[u]=cmx(s2[u],s1[fa],s2[fa],s1[u].se);
    if(s1[u].se!=f[u])ans[f[u]]=min(ans[f[u]],mkp(s1[u].fi+s[u]+a[u],s1[u].se));
    else if(s2[u].se!=f[u])ans[f[u]]=min(ans[f[u]],mkp(s2[u].fi+s[u]+a[u],s2[u].se));
    for(int i=hd[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa)continue;
        dfs2(v,u);
    }
    return;
}
signed main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
    for(int i=1,u,v,w;i<n;i++)scanf("%lld%lld%lld",&u,&v,&w),add(u,v,w),add(v,u,w);
    for(int i=1;i<=n;i++)f[i]=i;
    cnt=n;
    while(cnt>1)
    {
        dfs1(1,0);
        for(int i=1;i<=n;i++)ans[i]=mkp(1e18,-1);
        dfs2(1,0);
        for(int i=1;i<=n;i++)
        {
            if(f[i]==i&&ans[i].se>0&&find(i)!=find(ans[i].se))f[i]=ans[i].se,res+=ans[i].fi;
        }
        cnt=0;
        for(int i=1;i<=n;i++)cnt+=(find(i)==i);
    }
    printf("%lld",res);
    return 0;
}

标签:int,s2,s1,fa,boruvka,sum,se
From: https://www.cnblogs.com/DLYdly1105/p/18629215

相关文章

  • Boruvka求最小生成树
    写在前面这是我学这个算法的时候看的博客:推荐博客(除了这个还看了老师发的资料)为了复习以及加深理解,来简单写一篇学习笔记(预计半小时写完)关于Boruvka起因是在模拟赛遇到的T3。大意是:对于给出的完全图,\(w_{(u,v)}=a_{\max{(u,v)}}-a_{\min{(u,v)}}\),求最小生成树。烧烤了半场......
  • boruvka 算法学习笔记
    boruvka算法就是最小生成树B算法。B算法的思路是每次对每个连通块,求出它能连出去的权值最小的边,然后再按边权从小到大合并。由于每次操作连通块数至少减半,所以复杂度是\(O(m\logn)\)。1.CF1305GKuroniandAntihype题意:长为\(n\)的数列\(a\),现在要选择全部数,每一次你......
  • Boruvka 最小生成树
    Boruvka最小生成树一个用得相对少的最小生成树算法。需要注意的是只能做边权互不相同的问题。怎么做?首先先将所有点看做独立的连通块。然后对于每个联通块找到最小的一条出边,判了连通性,可以就直接合并就行了。这样每次联通块个数每次都会变为原来的\(\frac{1}{2}\),所以只......
  • Boruvka 算法简记
    这个算法怕是只会存在于模拟赛里了。Boruvka算法是用于解决完全图的生成树的一类算法,因为完全图边数很多,因此普通时间复杂度基于边数的做法不适用。Boruvka算法核心思想......
  • Educational Codeforces Round 32 G Xor tree Boruvka算法
    求一个n个点的完全图每条边的权值为两点之间的异或值求最小生成树。在完全图上做最小生成树一般都是Boruvka算法即每次每个点都找一个离自己最近的点合并这样最多合并lo......