【BZOJ】1977 [BeiJing2010组队]次小生成树 Tree kruskal+LCA

题目传送门

这题好像在好久以前就讲过啊……为什么我现在才去做……

其实正解挺好想的,首先做一棵最小生成树,然后枚举一条非树边,显然这条非树边的加入可以使原本树上的唯一路径形成一个环,把这个环上严格第二小的边删掉,就是一棵严格次小生成树了。然后对所有的次小生成树求一个最小权值即可。

至于怎么查找原来的唯一路径上的严格次大值,可以用ST表维护最大值和次大值。

附上AC代码:

#include 
#include 
#include 
using namespace std;

typedef long long ll;
const int M=3e5+10,N=1e5+10;
struct note{
    int x,y,w;
    bool operator < (const note lyf) const {return wstruct side{
    int to,w,nt;
}s[N<<1];
int n,m,fa[N],h[N],num,d[N],f[N][20],mx1[N][20],mx2[N][20],mn;
ll ans;
bool b[M];

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

inline int gf(int x){return x==fa[x]?x:fa[x]=gf(fa[x]);}

inline void add(int x,int y,int w){
    s[++num]=(side){y,w,h[x]},h[x]=num;
    s[++num]=(side){x,w,h[y]},h[y]=num;
}

inline void so(int x,int fa){
    for (int i=1; i<=17; ++i){
        if (d[x]<(1<break;
        f[x][i]=f[f[x][i-1]][i-1];
        mx1[x][i]=max(mx1[x][i-1],mx1[f[x][i-1]][i-1]);
        if (mx1[x][i-1]==mx1[f[x][i-1]][i-1]) mx2[x][i]=max(mx2[x][i-1],mx2[f[x][i-1]][i-1]);
        else {
            mx2[x][i]=min(mx1[x][i-1],mx1[f[x][i-1]][i-1]);
            mx2[x][i]=max(mx2[x][i],max(mx2[x][i-1],mx2[f[x][i-1]][i-1]));
        }
    }
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=fa) d[s[i].to]=d[f[s[i].to][0]=x]+1,mx1[s[i].to][0]=s[i].w,so(s[i].to,x);
}

inline int lca(int x,int y){
    if (d[x]int t=d[x]-d[y];
    for (int i=0; i<=17; ++i) if (t&(1<for (int i=17; i>=0; --i) if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    if (x==y) return x;
    return f[x][0];
}

inline void calc(int x,int fa,int w){
    int d1=0,d2=0,t=d[x]-d[fa];
    for (int i=0; i<=17; ++i)
        if (t&(1<if (mx1[x][i]>d1) d2=d1,d1=mx1[x][i];
            d2=max(d2,mx2[x][i]),x=f[x][i];
        }
    if (d1!=w) mn=min(mn,w-d1); else mn=min(mn,w-d2);
    return;
}

int main(void){
    read(n),read(m);
    for (int i=1; i<=m; ++i) read(a[i].x),read(a[i].y),read(a[i].w);
    sort(a+1,a+1+m);
    for (int i=1; i<=n; ++i) fa[i]=i;
    for (int i=1; i<=m; ++i){
        int fx=gf(a[i].x),fy=gf(a[i].y);
        if (fx!=fy) fa[fx]=fy,ans+=a[i].w,b[i]=1,add(a[i].x,a[i].y,a[i].w);
    }
    so(1,0),mn=2e9;
    for (int i=1; i<=m; ++i)
        if (!b[i]){
            int t=lca(a[i].x,a[i].y);
            calc(a[i].x,t,a[i].w),calc(a[i].y,t,a[i].w);
        }
    return printf("%lld\n",ans+mn),0;
}

你可能感兴趣的:(BZOJ,LCA,最小生成树,并查集,水题)