bzoj1977: [BeiJing2010组队]次小生成树 Tree

1977: [BeiJing2010组队]次小生成树 Tree

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 3863  Solved: 1107
[Submit][Status][Discuss]

Description

小 C 最近学了很多最小生成树的算法，Prim 算法、Kurskal 算法、消圈算法等等。 正当小 C 洋洋得意之时，小 P 又来泼小 C 冷水了。小 P 说，让小 C 求出一个无向图的次小生成树，而且这个次小生成树还得是严格次小的，也就是说： 如果最小生成树选择的边集是 EM，严格次小生成树选择的边集是 ES，那么需要满足：(value(e) 表示边 e的权值)  这下小 C 蒙了，他找到了你，希望你帮他解决这个问题。

Input

第一行包含两个整数N 和M，表示无向图的点数与边数。 接下来 M行，每行 3个数x y z 表示，点 x 和点y之间有一条边，边的权值为z。

Output

包含一行，仅一个数，表示严格次小生成树的边权和。(数据保证必定存在严格次小生成树)

Sample Input

5 6
1 2 1
1 3 2
2 4 3
3 5 4
3 4 3
4 5 6

Sample Output

11

HINT

 

数据中无向图无自环； 50% 的数据N≤2 000 M≤3 000； 80% 的数据N≤50 000 M≤100 000； 100% 的数据N≤100 000 M≤300 000 ，边权值非负且不超过 10^9 。

 

Source

 

[Submit][Status][Discuss]


HOME Back

 

 

题解：

次小生成树一定是最小生成树的基础上改变了一条边，那么我们就枚举改变的那条边a——b，然后在最小生成树上跳a和b的lca，再在路径上找出一条最大的边换成加入的边即可（因为加入一条边之后会在a,b到lca的路径上产生一个环）。

 

代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
struct aaa{
	ll fa,mx,cmx,l,r,v;
}fa[100001][21],a[500001];
ll tot,n1,n,m,ans,sum,e[500001],jia[500001],head[500001],next[500001],dep[500001],f[500001],g[500001];
bool cmp(aaa a,aaa b){
	return a.v<b.v;
}
void build(ll t,ll k,ll s){
	tot++;
	e[tot]=k;jia[tot]=s;
	next[tot]=head[t];head[t]=tot;
}
void dfs(ll x){
	ll i;
	for(i=1;(1<<i)<=dep[x];i++){
		fa[x][i].fa=fa[fa[x][i-1].fa][i-1].fa;
		fa[x][i].mx=max(fa[fa[x][i-1].fa][i-1].mx,fa[x][i-1].mx);
		fa[x][i].cmx=max(fa[fa[x][i-1].fa][i-1].cmx,fa[x][i-1].cmx);
		if(fa[x][i-1].mx<fa[fa[x][i-1].fa][i-1].mx&&fa[x][i-1].mx>fa[x][i].cmx)fa[x][i].cmx=fa[x][i-1].mx;
		 else if(fa[x][i-1].mx>fa[fa[x][i-1].fa][i-1].mx&&fa[fa[x][i-1].fa][i-1].mx>fa[x][i].cmx)fa[x][i].cmx=fa[fa[x][i-1].fa][i-1].mx;
	}
	for(i=head[x];i;i=next[i])
	 if(e[i]!=fa[x][0].fa){
	 	dep[e[i]]=dep[x]+1;
	 	fa[e[i]][0].fa=x;
	 	fa[e[i]][0].mx=jia[i];
	 	dfs(e[i]);
	 }
}
ll lca(ll x,ll y){
	ll i,t;
	if(dep[x]<dep[y])swap(x,y);
	t=dep[x]-dep[y];
	for(i=0;i<=20;i++)
	 if(t&(1<<i))x=fa[x][i].fa;
	if(x==y)return x;
	for(i=20;~i;i--)
	 if(fa[x][i].fa!=fa[y][i].fa){
	 	x=fa[x][i].fa;y=fa[y][i].fa;
	 }
	return fa[x][0].fa; 
}
void solve(ll x,ll y,ll val){
	ll i,mx=0,cmx=0,t;
	//printf("%lld %lld %lld %lld\n",x,y,dep[x],dep[y]);
	t=dep[x]-dep[y];
	for(i=0;i<=20;i++)
	 if(t&(1<<i)){
	 	//printf("%lld\n",i);
	 	if(fa[x][i].mx>mx){
	 		cmx=mx;
	 		mx=fa[x][i].mx;
		 }
		 cmx=max(cmx,fa[x][i].cmx);
		 x=fa[x][i].fa;
	 }
	//printf("%lld %lld\n",mx,cmx); 
	if(mx==val)sum=min(sum,val-cmx);
	 else sum=min(sum,val-mx); 
}
ll find(ll t){
	if(t==f[t])return t;
	return f[t]=find(f[t]);
}
int main(){
	ll i,x,y;
	scanf("%lld%lld",&n,&m);
	for(i=1;i<=m;i++)scanf("%lld%lld%lld",&a[i].l,&a[i].r,&a[i].v);
	sort(a+1,a+m+1,cmp);
	for(i=1;i<=n;i++)f[i]=i;
	for(i=1;i<=m;i++){
		if(n1==n-1)break;
		x=find(a[i].l);y=find(a[i].r);
		if(x!=y){
			f[x]=y;n1++;
			ans+=a[i].v;
			g[i]=1;
			build(a[i].l,a[i].r,a[i].v);
			build(a[i].r,a[i].l,a[i].v);
		}
	}
	//dep[1]=1;
	sum=1e9;
	dfs(1);
	for(i=1;i<=m;i++)
	 if(!g[i]){
	 	x=lca(a[i].l,a[i].r);
	 	//printf("%lld %lld %lld %lld\n",i,x,a[i].l,a[i].r);
	 	solve(a[i].l,x,a[i].v);
	 	solve(a[i].r,x,a[i].v);
	 }
	//printf("%lld\n",sum); 
	printf("%lld",ans+sum); 
}