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
Data Constraint
30% n<=1000
100% n,m<=200000
分析
//为什么题目名是这样的,我也不知道
考虑先对原图求一个最小生成树。对于每一个询问,显然除了必须选的边外,其它边都是最小生成树上的边是最优的。
那么应该删掉最小生成树上的哪条边呢?
这条边一定满足:删去之后再加上必选的边,依然构成一棵树。
设必选的边连接(u,v),那么这条边就在u到v的简单路径上。所以求出该路径上的最大权值的边,就是应该删的了。
时间复杂度 O(mlogn)
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=200005,maxm=400005,Log=17;
typedef long long LL;
int n,m,h[maxn],e[maxm],next[maxm],len[maxm],E[maxn][3],tot,fa[maxn][Log+1],g[maxn][Log+1],a[maxn],f[maxn],dep[maxn];
LL ans;
bool cmp(int x,int y)
{
return E[x][2]<E[y][2];
}
int get(int x)
{
return (!f[x])?x:f[x]=get(f[x]);
}
void add(int x,int y,int l)
{
e[++tot]=y; next[tot]=h[x]; len[tot]=l; h[x]=tot;
}
void init(int x)
{
dep[x]=dep[fa[x][0]]+1;
for (int i=h[x];i;i=next[i]) if (e[i]!=fa[x][0])
{
fa[e[i]][0]=x; g[e[i]][0]=len[i]; init(e[i]);
}
}
int getlca(int x,int y)
{
if (dep[x]<dep[y]) x^=y^=x^=y;
for (int i=Log;i>=0;i--) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
for (int i=Log;i>=0;i--) if (fa[x][i]!=fa[y][i])
{
x=fa[x][i]; y=fa[y][i];
}
return (x!=y)?fa[x][0]:x;
}
int maxlen(int x,int y)
{
int lca=getlca(x,y),M=0;
for (int i=Log;i>=0;i--) if (dep[fa[x][i]]>=dep[lca])
{
M=max(M,g[x][i]); x=fa[x][i];
}
for (int i=Log;i>=0;i--) if (dep[fa[y][i]]>=dep[lca])
{
M=max(M,g[y][i]); y=fa[y][i];
}
return M;
}
int main()
{
freopen("street.in","r",stdin); freopen("street.out","w",stdout);
scanf("%d%d",&n,&m);
for (int i=0;i<m;i++) scanf("%d%d%d",&E[a[i]=i][0],&E[i][1],&E[i][2]);
sort(a,a+m,cmp);
for (int i=0;i<m;i++)
{
int x=get(E[a[i]][0]),y=get(E[a[i]][1]);
if (x!=y)
{
f[y]=x;
ans+=E[a[i]][2];
add(E[a[i]][0],E[a[i]][1],E[a[i]][2]); add(E[a[i]][1],E[a[i]][0],E[a[i]][2]);
}
}
init(1);
for (int j=1;j<=Log;j++)
for (int i=1;i<=n;i++)
{
fa[i][j]=fa[fa[i][j-1]][j-1];
g[i][j]=max(g[i][j-1],g[fa[i][j-1]][j-1]);
}
for (int i=0;i<m;i++) printf("%lld\n",ans+E[i][2]-maxlen(E[i][0],E[i][1]));
fclose(stdin); fclose(stdout);
return 0;
}