【NOIP2016模拟7.8】Stree

Description

给出n个点,m条有权边,现对于每一条边，你需要回答出包含这条边的最小生成树的总边权值。
Input
第一行两个数n,m
接下来m行i,j,k,表示i与j间有一条权值为k的边
Output
m行答案
Sample Input
5 7
1 2 3
1 3 1
1 4 5
2 3 2
2 5 3
3 4 2
4 5 4
Sample Output
9
8
11
8
8
8
9

The solution

看到最小生成树，一言不合直接上克鲁斯卡尔。(╯3╰)
对于最小生成树上的边答案显然是生成树边权和。
只需求该边两边的点之间最短路径上权值最大的边，用lca即可。

即：

先构出最小生成树，对于在MST里面的边答案就是最小生成树权和，
对于其他边i ,j 只要在最小生成树上找出两点路径上最大的边即可
可用树上倍增。

涉及算法：lca，树上倍增，最小生成树

Code

#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define fd(i,a,b) for (int i=a;i>=b;i--)
#define N 200005 
int n,m,l=0,t[N*2],next[N*2],last[N],Dad[N],deep[N],F[N][20];
long long Max,Value=0,a[N*2],G[N][20],Ans[N*2];
bool bz[N*2];
struct Re
{
    int x,y,id;
    long long z;
}b[N];
using namespace std;
bool cmp(Re a,Re b)
{
    return a.z<b.z;
}
void add(int x,int y,long long z)
{
    t[++l]=y;
    next[l]=last[x];
    last[x]=l;
    a[l]=z;
}
void dfs(int x,int fa,int dep)
{
    F[x][0]=fa;
    deep[x]=dep;
    for(int k=last[x];k;k=next[k])
    {
        if (t[k]==fa) continue;
        G[t[k]][0]=a[k];
        dfs(t[k],x,dep+1);
    }
}
int get(int x){ return !Dad[x]?x:(Dad[x]=get(Dad[x]));}
void Jump(int &x,int i){Max=max(Max,G[x][i]),x=F[x][i]; }
long long Lca(int x,int y)
{
    Max=0;
    if (deep[x]<deep[y]) swap(x,y);
    fd(i,int(log2(deep[x])),0) if (deep[F[x][i]]>=deep[y]) Jump(x,i);
    fd(i,int(log2(deep[x])),0) if (F[x][i]!=F[y][i]) Jump(x,i),Jump(y,i);
    if (x!=y) Max=max(Max,max(G[x][0],G[y][0]));
    return Max; 
}
int main()
{
    freopen("street.in","r",stdin);
    freopen("street.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,m) scanf("%d%d%lld",&b[i].x,&b[i].y,&b[i].z),b[i].id=i;
    sort(b+1,b+m+1,cmp);
    int tot=0;
    fo(i,1,m)
    {
        int xx=get(b[i].x);
        int yy=get(b[i].y);
        if (xx!=yy)
        {
            Dad[xx]=yy;
            Value+=b[i].z;
            bz[i]=true;
            add(b[i].x,b[i].y,b[i].z);
            add(b[i].y,b[i].x,b[i].z);
            if (++tot>=n-1) break;
        }
    }
    dfs(1,0,1);
    fo(j,1,int(log2(n)))
        fo(i,1,n)
        {
            F[i][j]=F[F[i][j-1]][j-1];
            G[i][j]=max(G[i][j-1],G[F[i][j-1]][j-1]);
        }
    fo(i,1,m)
        if (bz[i]) Ans[b[i].id]=Value;
        else Ans[b[i].id]=Value+b[i].z-Lca(b[i].x,b[i].y);
    fo(i,1,m) printf("%lld\n",Ans[i]);
    return 0;   
}