P4180 [BJWC2010] 严格次小生成树

                                        P4180 [BJWC2010] 严格次小生成树

[BJWC2010] 严格次小生成树 - 洛谷

 思路：那么我们首先需要求出最小生成树来，同时我们建立其最小生成树的图，并且得到最小生成树的权值和为sum，然后我们的思想是枚举所有非树边，那么我们在添加了一条非树边之后就会形成一个环，所以我们要再删除一条边，删除的边权越大得到的生成树越小，所以我们可以用lca维护任意两个点之间的路径上的最大边权与次大边权，那么我们对于a,b两个点，我们可以看一下a,b之间路径上的最大边权是不是等于c，如果等于c则我们用删去次大边权，因为要求是严格次小生成树，否则则用最大边权去更新，最后枚举所有的非树边统计出最小值即可

#include<iostream>
#include<cstring>
#include<string>
#include<sstream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<vector> 
#include<set>
#include<unordered_map>
#include<ctime>
#include<cstdlib>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> PII;
typedef pair<int,pair<int,int> > PIII;
const double eps=1e-7;
const int N=5e5+7 ,M=1e6+7, INF=0x3f3f3f3f,mod=1e9+7;
const long long int llINF=0x3f3f3f3f3f3f3f3f;
inline ll read() {ll x=0,f=1;char c=getchar();while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {x=(ll)x*10+c-'0';c=getchar();} return x*f;}
inline void write(ll x) {if(x < 0) {putchar('-'); x = -x;}if(x >= 10) write(x / 10);putchar(x % 10 + '0');}
inline void write(ll x,char ch) {write(x);putchar(ch);}
void stin() {freopen("in_put.txt","r",stdin);freopen("my_out_put.txt","w",stdout);}
bool cmp0(int a,int b) {return a>b;}
template<typename T> T gcd(T a,T b) {return b==0?a:gcd(b,a%b);}
template<typename T> T lcm(T a,T b) {return a*b/gcd(a,b);}
void hack() {printf("\n----------------------------------\n");}

int T,hackT;
int n,m,k;
int h[N],e[M],ne[M],w[M],idx;
struct Edge{
    int a,b,c;
    bool operator < (const Edge &k) const {
        return c<k.c;
    }
};
Edge edge[M];
int ff[N];
int f[N][21],d1[N][21],d2[N][21];
int dep[N];
bool st[M];

void add(int a,int b,int c) {
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}

int find(int x) {
    if(x!=ff[x]) ff[x]=find(ff[x]);
    return ff[x];
}

void change(int &a,int &b,int c) {
    if(c>a) {
        b=a;
        a=c;
    }else if(c!=a&&c>b) b=c;
}

void bfs(int u) {
    memset(dep,0x3f,sizeof dep);
    dep[0]=0,dep[u]=1;
    queue<int> q;
    q.push(u);
    
    while(q.size()) {
        auto it=q.front();
        q.pop();
        
        for(int i=h[it];i!=-1;i=ne[i]) {
            int j=e[i];
            
            if(dep[j]>dep[it]+1) {
                dep[j]=dep[it]+1;
                q.push(j);
                f[j][0]=it;
                d1[j][0]=w[i];
                d2[j][0]=-INF;
                for(int k=1;k<=20;k++) {
                    int anc=f[j][k-1];
                    f[j][k]=f[anc][k-1];
                    d1[j][k]=d2[j][k]=-INF;
                    change(d1[j][k],d2[j][k],d1[j][k-1]);
                    change(d1[j][k],d2[j][k],d2[j][k-1]);
                    change(d1[j][k],d2[j][k],d1[anc][k-1]);
                    change(d1[j][k],d2[j][k],d2[anc][k-1]);
                }
            }
        }
    }
}

int lca(int a,int b,int c) {
    if(dep[a]<dep[b]) swap(a,b);
    int x=0,y=0;
    for(int i=20;i>=0;i--) {
        if(dep[f[a][i]]>=dep[b]) {
            change(x,y,d1[a][i]);
            change(x,y,d2[a][i]);
            a=f[a][i];
        }
    }
    
    if(a==b) {
        if(c>x) return c-x;
        if(c>y) return c-y;
        return INF;
    } 
    for(int i=20;i>=0;i--) {
        if(f[a][i]!=f[b][i]) {
            change(x,y,d1[a][i]);
            change(x,y,d2[a][i]);
            change(x,y,d1[b][i]);
            change(x,y,d2[b][i]);
            a=f[a][i];
            b=f[b][i];
        }
    }
    
    change(x,y,d1[a][0]);
    change(x,y,d1[b][0]);
    
    if(c>x) return c-x;
    if(c>y) return c-y;
    return INF;
}

void solve() {
	n=read(),m=read();
	
	memset(h,-1,sizeof h);
	for(int i=1;i<=n;i++) ff[i]=i;
	for(int i=1;i<=m;i++) {
	    int a=read(),b=read(),c=read();
	    edge[i]={a,b,c};
	}
	
	sort(edge+1,edge+1+m);
	
	ll sum=0;
	for(int i=1;i<=m;i++) {
	    int a=edge[i].a,b=edge[i].b,c=edge[i].c;
	    int fa=find(a),fb=find(b);
	    if(fa!=fb) {
	        add(a,b,c),add(b,a,c);
	        ff[fa]=ff[fb];
	        sum+=c;
	        st[i]=true;
	    }
	}
	
	bfs(1);
	
	ll res=llINF;
	for(int i=1;i<=m;i++) {
	    if(st[i]) continue;
	    int a=edge[i].a,b=edge[i].b,c=edge[i].c;
	    if(a==b) continue;
	    res=min(res,sum+lca(a,b,c));
	}
	
	printf("%lld\n",res);
}   

int main() {
    // init();
    // stin();

    // scanf("%d",&T);
    T=1; 
    while(T--) hackT++,solve();
    
    return 0;       
}