[Educational Round 3][Codeforces 609E. Minimum spanning tree for each edge]

这题原本是想放在educational round 3的题解里的，但以为颇有意思就单独拿出来写了

题目连接：609E - Minimum spanning tree for each edge

题目大意：n个点，m条边，对每条边，询问包含此边的最小生成树的边权之和

题解：大部分人都是用LCA写的，这里提供一个更为精妙的作法。

　　　模拟Kruskal算法建MST的过程，先将m条边按边权排序，依次进行判断。若点对(u,v)属于同一个连通块，则加入边{u,v,w}后会造成一个环，把环中最大的边换成w会多产生的权重就是包含这条边的MST比原先的MST多出来的权重，不然在MST中加入这条边。

　　　而这里加边的时候并非选择直接在(u,v)之间连边，而是在他们的祖先之间连边，并有以下判断：若点v的子树大小大于点u的子树大小，则把v看作是u的父亲（将u所在的子树并入v），反之亦然。判断加边后是否会成环时，则只需暴力一层层寻找当前点的父亲便可，便是用O(n)的方法求LCA，但因为以前建树的方式用到了[small to large]的思想（就是启发式合并），因此往上爬的层数不会超过log n。所以时间复杂度为O(mlog n)。

#include<bits/stdc++.h>
using namespace std; #define N 200001
#define LL long long
struct rua{LL u,v,w,id;}a[N]; LL n,m,mst,fa[N],f[N],sz[N],l[N]; bool cmp(rua x,rua y){return x.w<y.w;} LL add(LL u,LL v,LL w) { LL mx=0; while((fa[u]!=u ||fa[v]!=v) && u!=v) if(fa[u]==u || (fa[v]!=v && sz[v]<=sz[u])) mx=max(mx,l[v]),v=fa[v]; else mx=max(mx,l[u]),u=fa[u]; if(u==v)return w-mx; if(sz[u]<sz[v])swap(u,v); sz[u]+=sz[v],fa[v]=u,l[v]=w; return mst+=w,0; } int main() { scanf("%I64d%I64d",&n,&m); for(int i=1;i<=m;i++) scanf("%I64d%I64d%I64d",&a[i].u,&a[i].v,&a[i].w),sz[i]=1,fa[i]=a[i].id=i; sort(a+1,a+m+1,cmp); for(int i=1;i<=m;i++) f[a[i].id]=add(a[i].u,a[i].v,a[i].w); for(int i=1;i<=m;i++) printf("%I64d\n",mst+f[i]); }

View Code