[BJWC2010]严格次小生成树

板子题

先kruskal得到最小生成树，再枚举未加进生成树中的边i，该边i和生成树树形成了一个环，选择这个环上最大的不等于i的权值的边去掉就得到一个新的树，所有这样得到的树权值之和即为次小生成树。

求环上最大边可以用倍增或者树链剖分，由于最大边可能和才加上去的边相等，所以还要维护一个次大边

#include<bits/stdc++.h>
#define N 100005
#define M 300005
using namespace std;
typedef long long ll;
const ll INF = 10000000000000000;
int n,m,fa[N],f[N][20],dep[N];
bool vis[M];
ll maxx[N][20],naxx[N][20],ans,minn;
struct E {int u,v;ll w;} e[M];
struct Edge {int next,to;ll dis;}edge[M<<1];
int head[N],cnt=1;
void add_edge(int from,int to,ll dis)
{
    edge[++cnt].next=head[from];
    edge[cnt].to=to;
    edge[cnt].dis=dis;
    head[from]=cnt;
}
template <class T>
void read(T &x)
{
    char c;int sign=1;
    while((c=getchar())>'9'||c<'0') if(c=='-') sign=-1; x=c-48;
    while((c=getchar())>='0'&&c<='9') x=(x<<1)+(x<<3)+c-48; x*=sign;
}
bool cmp(E a,E b) {return a.w<b.w;}
int find(int x) {return x==fa[x] ? x : fa[x]=find(fa[x]);}
void kruskal()
{
    sort(e+1,e+m+1,cmp);
    int cnt=0; minn=0;
    for(int i=1;i<=m;++i)
    {
        int fu=find(e[i].u),fv=find(e[i].v);
        if(fu!=fv)
        {
            fa[fu]=fv;
            add_edge(e[i].u,e[i].v,e[i].w);
            add_edge(e[i].v,e[i].u,e[i].w);
            vis[i]=1;
            minn+=e[i].w;
            if(++cnt==n-1) break;
        }
    }
}
void dfs(int rt,int fa)
{
    dep[rt]=dep[fa]+1;
    for(int i=head[rt];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa) continue;
        maxx[v][0]=edge[i].dis;
        naxx[v][0]=0;
        f[v][0]=rt;
        for(int i=1;i<20;++i)
        {
            f[v][i]=f[f[v][i-1]][i-1];
            maxx[v][i]=max(maxx[v][i-1],maxx[f[v][i-1]][i-1]);
            naxx[v][i]=max(naxx[v][i-1],naxx[f[v][i-1]][i-1]);
            if(maxx[v][i-1]!=maxx[v][i]) naxx[v][i]=max(naxx[v][i],maxx[v][i-1]);
            if(maxx[f[v][i-1]][i-1]!=maxx[v][i]) naxx[v][i]=max(naxx[v][i],maxx[f[v][i-1]][i-1]);
        }
        dfs(v,rt);
    }
}
ll query(int x,int y,ll MAX)
{
    ll ret=-INF;
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=19;i>=0;--i)
    {
        if(dep[f[x][i]]>=dep[y])
        {
            if(maxx[x][i]!=MAX) ret=max(ret,maxx[x][i]);
            else ret=max(ret,naxx[x][i]);
            x=f[x][i];
        }
    }
    if(x==y) return ret;
    for(int i=19;i>=0;--i)
    {
        if(f[x][i]!=f[y][i])
        {
            if(maxx[x][i]!=MAX) ret=max(ret,maxx[x][i]);
            else ret=max(ret,naxx[x][i]);
            if(maxx[y][i]!=MAX) ret=max(ret,maxx[y][i]);
            else ret=max(ret,naxx[y][i]);
            x=f[x][i];y=f[y][i];
        }
    }
    if(maxx[x][0]!=MAX) ret=max(ret,maxx[x][0]); else ret=max(ret,naxx[x][0]);
    if(maxx[y][0]!=MAX) ret=max(ret,maxx[y][0]); else ret=max(ret,naxx[y][0]);
    return ret;
}
int main()
{
    read(n);read(m);
    for(int i=1;i<=n;++i) fa[i]=i;
    for(int i=1;i<=m;++i) read(e[i].u),read(e[i].v),read(e[i].w);
    kruskal();
    ans=INF;
    dfs(1,0);
    for(int i=1;i<=m;++i)
    {
        if(vis[i]) continue;
        ll q=query(e[i].u,e[i].v,e[i].w);
        if(minn-q+e[i].w!=minn) ans=min(ans,minn-q+e[i].w);
    }
    cout<<ans<<endl;
    return 0;
}

转载于:https://www.cnblogs.com/Chtholly/p/11170161.html