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]);
}