Stree

题目

给出n个点,m条有权边,现对于每一条边，你需要回答出包含这条边的最小生成树的总边权值。

分析

首先我们可以构造一个对于这n个点,m条有权边的最小生成树，显然，这是一棵最小的生成树。
那么这棵生成树的边的答案就是这棵生成树的总边权。
然后，就要考虑这棵生成树的其他边了。
在这棵生成树上，如果我们给它加一条新的边，那么，一定会形成一个环。
所以，我们把这个环中最大的边（当然不是新加入的边啦）删掉，这棵新树的总边权就是答案。
怎样删掉这个环中最大的边呢？
发现，实际上就是删掉加入的新边的两个顶点在原树上的路径上最大的边，那么就可以打倍增lca找到最大的边。（想打树链剖分也可以，不拦你）

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
const long long maxlongint=2147483647;
const long long mo=1000000007;
const long long N=200006;
using namespace std;
long long ans[N],way[N][3],n,m,fa[N],sum,bef[N],last[N*2],next[N*2],to[N*2],f[N][20],g[N][20],tot,v[N*2],deep[N];
void bj(long long x,long long y,long long z)
{
	next[++tot]=last[x];
	last[x]=tot;
	to[tot]=y;
	v[tot]=z;
}
void q(long long l,long long r)
{
	long long i=l,j=r,mid=way[(l+r)/2][0],e;
	while(i<j)
	{
		while(way[i][0]<mid) i++;
		while(way[j][0]>mid) j--;
		if(i<=j)
		{
			e=way[i][0];
			way[i][0]=way[j][0];
			way[j][0]=e;
			e=way[i][1];
			way[i][1]=way[j][1];
			way[j][1]=e;
			e=way[i][2];
			way[i][2]=way[j][2];
			way[j][2]=e;
			e=bef[j];
			bef[j]=bef[i];
			bef[i]=e;
			i++;
			j--;
		}
	}
	if(i<r) q(i,r);
	if(l<j) q(l,j);
}
long long get(long long x)
{
	if(fa[x]==x) return x;
	fa[x]=get(fa[x]);
	return fa[x];
}
void dg(long long x,long long y)
{
	for(long long i=last[x];i;i=next[i])
	{
		long long j=to[i];
		if(j!=y)
		{
			deep[j]=deep[x]+1;
			g[j][0]=x;
			f[j][0]=v[i];
			dg(j,x);
		}
	}
}	
long long lca(long long x,long long y)
{
	long long l;
    if(deep[x]<=deep[y])
    {
        l=x;
        x=y;
        y=l;
    }
    long long p=0;
    for(long long i=log2(n);i>=0;i--)
    {
    	if(deep[g[x][i]]>=deep[y])
    	{
    		p=max(p,f[x][i]);
    		x=g[x][i];
		}
	}
    for(long long i=log2(n);i>=0;i--)
    {
    	if(g[x][i]!=g[y][i])
    	{
    		p=max(p,max(f[x][i],f[y][i]));
    		x=g[x][i];
    		y=g[y][i];
		}
	}
    if(x!=y) p=max(p,max(f[x][0],f[y][0]));
	return p;
}
int main()
{
	freopen("street.in","r",stdin);
	freopen("street.out","w",stdout);
	scanf("%lld%lld",&n,&m);
	for(long long i=1;i<=m;i++)
	{
		scanf("%lld%lld%lld",&way[i][1],&way[i][2],&way[i][0]);
		fa[i]=i;
		bef[i]=i;
	}
	q(1,m);
	long long k=0;
	sum=0;
	for(long long i=1;i<=m,k<n-1;i++)
	{
		long long x=get(way[i][1]);
		long long y=get(way[i][2]);
		if(x==y) continue;
		k++;
		ans[bef[i]]=1;
		bj(way[i][1],way[i][2],way[i][0]);
		bj(way[i][2],way[i][1],way[i][0]);
		sum+=way[i][0];
		fa[x]=y;
	}
	deep[1]=1;
	dg(1,0);
	for(long long i=1;i<=log2(n);i++)
		for(long long j=1;j<=n;j++)
		{
			g[j][i]=g[g[j][i-1]][i-1];
			f[j][i]=max(f[j][i-1],f[g[j][i-1]][i-1]);
		}
	for(long long i=1;i<=m;i++)
	{
		if(!ans[bef[i]])
		{
			ans[bef[i]]=sum-lca(way[i][1],way[i][2])+way[i][0];
		}
		else 
			ans[bef[i]]=sum;
	}
	for(long long i=1;i<=m;i++)
		printf("%lld\n",ans[i]);
}