Codeforces609 E - Minimum spanning tree for each edge（最小生成树+LCA）

题意：

给定n个点m条边的带权无向图，保证图连通
对于每条边，计算包含它的最小生成树值

数据范围：n,m<=2e5

解法：

经典题，
先计算出最小生成树
对于非树边(u,v,w)，在最小生成树中加入这条边之后会形成环，
删掉u和v之间最大的边之后就是包含(u,v,w)的最小生成树
u和v之间最大的边可以用LCA来求

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define PI pair<int,int>
const int maxm=2e5+5;
struct Node{
    int a,b,c;
    int id;
}e[maxm];
int pre[maxm];
int n,m;
int ffind(int x){
    return pre[x]==x?x:pre[x]=ffind(pre[x]);
}
bool cmp(Node a,Node b){
    return a.c<b.c;
}
//
vector<PI>g[maxm];
int d[maxm];
int f[maxm][25];
int dist[maxm][25];
const int maxd=23;
void dfs(int x,int fa){
    for(int j=1;j<=maxd;j++){
        f[x][j]=f[f[x][j-1]][j-1];
        dist[x][j]=max(dist[x][j-1],dist[f[x][j-1]][j-1]);
    }
    for(auto i:g[x]){
        int v=i.first;
        if(v==fa)continue;
        d[v]=d[x]+1;
        dist[v][0]=i.second;
        f[v][0]=x;
        dfs(v,x);
    }
}
int cal(int a,int b){
    int ans=0;
    if(d[a]<d[b])swap(a,b);
    for(int i=maxd;i>=0;i--){
        if(d[f[a][i]]>=d[b]){
            ans=max(ans,dist[a][i]);
            a=f[a][i];
        }
    }
    if(a==b)return ans;
    for(int i=maxd;i>=0;i--){
        if(f[a][i]!=f[b][i]){
            ans=max(ans,dist[a][i]);
            ans=max(ans,dist[b][i]);
            a=f[a][i];
            b=f[b][i];
        }
    }
    ans=max(ans,dist[a][0]);
    ans=max(ans,dist[b][0]);
    return ans;
}
//
int mark[maxm];
int ans[maxm];
//
signed main(){
    ios::sync_with_stdio(0);
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>e[i].a>>e[i].b>>e[i].c;
        e[i].id=i;
    }
    //kruskal
    int sum=0;
    sort(e+1,e+1+m,cmp);
    for(int i=1;i<=n;i++){
        pre[i]=i;
    }
    for(int i=1;i<=m;i++){
        int x=ffind(e[i].a),y=ffind(e[i].b);
        if(x!=y){
            pre[x]=y;
            sum+=e[i].c;
            mark[e[i].id]=1;
            g[e[i].a].push_back({e[i].b,e[i].c});
            g[e[i].b].push_back({e[i].a,e[i].c});
        }
    }
    //
    d[1]=1;
    dist[1][0]=0;
    dfs(1,1);
    //
    for(int i=1;i<=m;i++){
        if(mark[e[i].id]){//树边
            ans[e[i].id]=sum;
        }else{//非树边
            ans[e[i].id]=sum-cal(e[i].a,e[i].b)+e[i].c;
        }
    }
    for(int i=1;i<=m;i++){
        cout<<ans[i]<<endl;
    }
    return 0;
}

---