AtCoder

Luogu翻译

分析

首先这个连边的方式很毒瘤,要想办法改一下。

可以发现,连了$(a,b)$、$(a+1,b)$、$(a+1,b+1)$三条边,与连了$(a,b)$、$(a,a+1)$、$(b,b+1)$三条边是等价的。

于是每次加边就变成加了一条边和一个环。

先来考虑环上的权值,设一个 $f[i]$ 表示 $i-1$ 到 $i$ 的最小边权。

显然找到一个环上初始的 $f$ 最小的点,再绕环一周更新即可。

然后把其它的 $Q$ 条边丢进去,求最小生成树即可。

上面这么说不是很清楚,可以画图理解一下。

细节见代码。

代码

//It is made by M_sea
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#define re register
typedef long long ll;
using namespace std;

inline void chkmin(int& x,int y) { if (y<x) x=y; }
inline int read() {
    int X=0,w=1; char c=getchar();
    while (c<'0'||c>'9') { if (c=='-') w=-1; c=getchar(); }
    while (c>='0'&&c<='9') X=X*10+c-'0',c=getchar();
    return X*w;
}

const int N=400000+10;

int n,m,Q;
int f[N],fa[N];

inline int find(int x) { return x==fa[x]?x:fa[x]=find(fa[x]); }
inline void merge(int x,int y) {
    x=find(x),y=find(y);
    if (x!=y) fa[x]=y;
}

struct Edge { int u,v,w; } e[N];
bool operator <(Edge a,Edge b) { return a.w<b.w; }

int main() {
    n=read(),Q=read();
    memset(f,0x3f,sizeof(f));
    for (re int i=1;i<=Q;++i) {
        int a=read()+1,b=read()+1,c=read();
        chkmin(f[a%n+1],c+1),chkmin(f[b%n+1],c+2);
        e[++m]=(Edge){a,b,c};
    }
    int pos=1;
    for (re int i=2;i<=n;++i) if (f[i]<f[pos]) pos=i;
    for (re int i=pos%n+1,j=pos;i!=pos;j=i,i=i%n+1) chkmin(f[i],f[j]+2);
    for (re int i=1;i<n;++i) e[++m]=(Edge){i,i+1,f[i+1]};
    e[++m]=(Edge){n,1,f[1]};
    sort(e+1,e+m+1);
    for (re int i=1;i<=n;++i) fa[i]=i;
    int tot=0; ll ans=0;
    for (re int i=1;i<=m;++i) {
        int u=e[i].u,v=e[i].v,w=e[i].w;
        if (find(u)==find(v)) continue;
        ++tot,ans+=w,merge(u,v);
        if (tot==n-1) break;
    }
    printf("%lld\n",ans);
    return 0;
}
最后修改:2019 年 09 月 25 日 01 : 20 PM