CodeForces - 609E(最下生成树+lca)

E. Minimum spanning tree for each edge
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

Connected undirected weighted graph without self-loops and multiple edges is given. Graph contains n vertices and m edges.

For each edge (u, v) find the minimal possible weight of the spanning tree that contains the edge (u, v).

The weight of the spanning tree is the sum of weights of all edges included in spanning tree.

Input

First line contains two integers n and m (1 ≤ n ≤ 2·105, n - 1 ≤ m ≤ 2·105) — the number of vertices and edges in graph.

Each of the next m lines contains three integers ui, vi, wi (1 ≤ ui, vi ≤ n, ui ≠ vi, 1 ≤ wi ≤ 109) — the endpoints of the i-th edge and its weight.

Output

Print m lines. i-th line should contain the minimal possible weight of the spanning tree that contains i-th edge.

The edges are numbered from 1 to m in order of their appearing in input.

Examples
input
Copy
5 7
1 2 3
1 3 1
1 4 5
2 3 2
2 5 3
3 4 2
4 5 4
output
9
8
11
8
8
8
9

思路:先生成最小 生成树, 这时候如果这条边是最小生成树的组成边,那么生成树不变, 如果不是, 那就用这条边替换该环上的最大边,求最大边方法是用倍增法求公共祖先, 求公共祖先顺带着求出最大边。


#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=2e5+10;
typedef long long ll;
int n, m, F[maxn], num=0, head[maxn], depth[maxn], grand[maxn][25], gw[maxn][25], N, maxt[maxn][25];


struct n1{
	int u, v, w;
	bool operator < (const n1& tmp) const{
		return w<tmp.w;
	}
}a[maxn], b[maxn];//a,b存放所有边,b供最终输出用

struct n2{
	int to, w, next;//存放最小生成树
}e[3*maxn];

int _find(int x){
	return F[x]==x? x : F[x]=_find(F[x]);
}

void addEdg(int u, int v, int w){
	e[num].to=v;
	e[num].w=w;
	e[num].next=head[u];
	head[u]=num++;
}

ll Kruskal(){
	int cnt=0;
	ll ans=0;
	for(int i=1; i<=n; i++) F[i]=i;
	for(int i=1; i<=m; i++)
	{
		int x=_find(a[i].u), y=_find(a[i].v), w=a[i].w;
		if(x!=y)
		{
			
			addEdg(a[i].u, a[i].v, a[i].w);
			addEdg(a[i].v, a[i].u, a[i].w);//存双边
			F[x]=y;
			cnt++;
			ans+=w;
			if(cnt==n-1)
				break;
		}
	}
	return ans;
}

void dfs(int x, int pre){
	for(int i=1; i<=N; i++)
	{
		grand[x][i]=grand[grand[x][i-1]][i-1];//倍增求祖先
		maxt[x][i]=max(maxt[x][i-1], maxt[grand[x][i-1]][i-1]);//倍增求最大值
	}
	
	for(int i=head[x]; i!=-1; i=e[i].next)
	{
		if(e[i].to==pre) continue;
		grand[e[i].to][0]=x;//初始化祖先数组
		depth[e[i].to]=depth[x]+1;//树的深度
		maxt[e[i].to][0]=e[i].w;//初始化最大值数组
		dfs(e[i].to, x);
	}
}

void init(){
	N=floor(log(n+0.0)/log(2.0));//倍增次幂的上限
	depth[1]=0;
	dfs(1, 0);
}

int lca(int a, int b){
	if(depth[a]>depth[b]) swap(a, b);//保证a<=b
	int mx=0;//添加边组成环中的最大值(当然不算该边的最大值)
	for(int i=N; i>=0; i--)//二进制分解
	{
		if(depth[a]<depth[b] &&  depth[a] <=depth[grand[b][i]] )//求出depth[a]==depth[b]
		{
			mx=max(mx, maxt[b][i]);
			b=grand[b][i];
		}	
	}
	
	for(int i=N; i>=0; i--)//找公共祖先
	{
		if(grand[a][i]!=grand[b][i])
		{
			mx=max(mx, maxt[a][i]);
			mx=max(mx, maxt[b][i]);
			a=grand[a][i];
			b=grand[b][i];
		}
	}
	if(a!=b)//如果a是b的祖先那么就可以不用操作这一步
	{
		mx=max( mx, maxt[a][0]);
		mx=max( mx, maxt[b][0]);
	}
	return mx;
}

int main(){
	memset(head, -1, sizeof(head));
	scanf("%d%d", &n, &m);
	for(int i=1; i<=m; i++)
	{
		int u, v, w;
		scanf("%d%d%d", &u, &v, &w);
		a[i].u=b[i].u=u; a[i].v=b[i].v=v; a[i].w=b[i].w=w;
	}
	
	sort(a+1, a+1+m);
	
	ll ans=0;	
	
	ans=Kruskal();
	
	init();
	
	for(int i=1; i<=m; i++)
	{
		
		printf("%lld\n", ans-lca(b[i].u, b[i].v)+b[i].w);
	}
	
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值