【树链剖分+MST】CF609E

Problem - E - Codeforces

题意：

思路：

先把全局的MST求出来，然后对于一条边，如果它本来就在MST中，说明代价就是MST的权值和，否则它加入MST中，此时MST形成了环，我们把环中最大的那条边删掉，就相当于用这条边换掉了环中最大边，这样就是最小权值和

那么怎么求环中最大边，直接树剖即可

Code：

#include <bits/stdc++.h>

#define int long long

using namespace std;

const int mxn=4e5+10;
const int mxe=4e5+10;

struct ty{
	int to,next,w;
}edge[mxe<<2];

struct ty2{
	int val,add;
}tree[mxe<<2];

struct ty3{
	int u,v,w,id,Flag=0;
}e[mxe<<2];

int N,M,rt,P;
int op,u,v,w;
int tot=0,idx=0,answ=0;
int F[mxn],ans[mxn],p[mxn];
int head[mxn],sz[mxn],Fa[mxn],dep[mxn],id[mxn];
int nw[mxn],son[mxn],top[mxn];

void pushup(int rt){
	tree[rt].val=max(tree[rt<<1].val,tree[rt<<1|1].val);
}
int query(int rt,int l,int r,int x,int y){
	if(x<=l&&r<=y){
		return tree[rt].val;
	}
	int mid=l+r>>1;
	int res=0;
	if(x<=mid) res=max(res,query(rt<<1,l,mid,x,y));
	if(y>mid) res=max(res,query(rt<<1|1,mid+1,r,x,y));
	return res;
}
int query_path(int u,int v){
	int res=0;
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		res=max(res,query(1,1,N,id[top[u]],id[u]));
		u=Fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	res=max(res,query(1,1,N,id[u]+1,id[v]));
	return res;
}
void dfs2(int u,int t){
	id[u]=++idx;
	nw[idx]=p[u];
	top[u]=t;
	if(!son[u]) return;
	dfs2(son[u],t);
	for(int i=head[u];~i;i=edge[i].next){
		if(edge[i].to==Fa[u]||edge[i].to==son[u]) continue;
		dfs2(edge[i].to,edge[i].to);
	}
}
void dfs1(int u,int fa){
	Fa[u]=fa;
	dep[u]=dep[fa]+1;
	sz[u]=1;
	for(int i=head[u];~i;i=edge[i].next){
		if(edge[i].to==fa){
			p[u]=edge[i].w;
			continue;
		}
		dfs1(edge[i].to,u);
		sz[u]+=sz[edge[i].to];
		if(sz[son[u]]<sz[edge[i].to]) son[u]=edge[i].to;
	}
}
void add(int u,int v,int w){
	edge[tot].w=w;
	edge[tot].to=v;
	edge[tot].next=head[u];
	head[u]=tot++;
}
void G_init(){
	tot=0;
	for(int i=0;i<mxn;i++){
		head[i]=-1;
	}
}
void build(int rt,int l,int r){
	if(l==r){
		tree[rt].val=nw[l];
		return;
	}
	int mid=l+r>>1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
	pushup(rt);
}
bool cmp(ty3 x,ty3 y){
	return x.w<y.w;
}
int find(int x){
	return F[x]=(x==F[x])?x:find(F[x]);
}
void join(int u,int v){
	int f1=find(u),f2=find(v);
	if(f1!=f2){
		F[f1]=f2;
	}
}
void solve(){
	cin>>N>>M;
	G_init();
	for(int i=1;i<=N;i++) F[i]=i;
	for(int i=1;i<=M;i++){
		cin>>u>>v>>w;
		e[i]={u,v,w,i,0};
	}
	sort(e+1,e+1+M,cmp);
	for(int i=1;i<=M;i++){
		if(find(e[i].u)==find(e[i].v)) continue;
		join(e[i].u,e[i].v);
		answ+=e[i].w;
		e[i].Flag=1;
		add(e[i].u,e[i].v,e[i].w);
		add(e[i].v,e[i].u,e[i].w);
	}
	dfs1(1,0);
	dfs2(1,0);
	build(1,1,N);
	for(int i=1;i<=M;i++){
		ans[e[i].id]=answ+e[i].w-query_path(e[i].u,e[i].v);
	}
	for(int i=1;i<=M;i++) cout<<ans[i]<<'\n';
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int __=1;//cin>>__;
	while(__--)solve();return 0;
}