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

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

Time Limit: 10 Sec   Memory Limit: 512 MB
Submit: 3097   Solved: 782
[ 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] 题解： 先求出最小生成树 因为要求严格次小，枚举每一条不是树枝的边，统计出最小差值，然后加上最小生成树的总权值就可以了。 代码贴上：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define N 100001
#define M 300001
#define inf 0x7fffffff
#define ll long long
using namespace std;
int n,m,tot,cnt,mn=inf;
ll ans;
int f[N],head[N],deep[N],fa[N][17],d1[N][17],d2[N][17];
struct data{int x,y,v;bool sel;}a[M];
struct edge{int to,next,v;}e[N*2];
bool cmp(data a,data b)
{return a.v<b.v;}
void ins(int u,int v,int w)
{e[++cnt].to=v;e[cnt].next=head[u];e[cnt].v=w;head[u]=cnt;}
void insert(int u,int v,int w)
{ins(u,v,w);ins(v,u,w);}
int find(int x){return x==f[x]?x:find(f[x]);}
void dfs(int x,int f)
{
	for(int i=1;i<=16;i++)
	{
		if(deep[x]<(1<<i))break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
        // d1记录最大 d2记录严格次大 
        d1[x][i]=max(d1[x][i-1],d1[fa[x][i-1]][i-1]);
        if(d1[x][i-1]==d1[fa[x][i-1]][i-1])
		    d2[x][i]=max(d2[x][i-1],d2[fa[x][i-1]][i-1]);
        else 
		{
		    d2[x][i]=min(d1[x][i-1],d1[fa[x][i-1]][i-1]);
		    d2[x][i]=max(d2[x][i-1],d2[x][i]);
		    d2[x][i]=max(d2[x][i],d2[fa[x][i-1]][i-1]);
		}
    }
    for(int i=head[x];i;i=e[i].next)
    	if(e[i].to!=f)
    	{
    		fa[e[i].to][0]=x;
    		d1[e[i].to][0]=e[i].v;
    		deep[e[i].to]=deep[x]+1;
    		dfs(e[i].to,x);
    	}
}
int lca(int x,int y)
{
    if(deep[x]<deep[y])swap(x,y);
    int t=deep[x]-deep[y];
    for(int i=0;i<=16;i++)
        if((1<<i)&t)x=fa[x][i];
    for(int i=16;i>=0;i--)
    {
        if(fa[x][i]!=fa[y][i])
        {x=fa[x][i];y=fa[y][i];}
    }
    if(x==y)return x;
    return fa[x][0];
}
void cal(int x,int f,int v)
{
    int mx1=0,mx2=0;
    int t=deep[x]-deep[f];
    for(int i=0;i<=16;i++)
    {
        if(t&(1<<i))
        {
           if(d1[x][i]>mx1)
           {
           	   mx2=mx1;
           	   mx1=d1[x][i];
           }
           mx2=max(mx2,d2[x][i]);
           x=fa[x][i];
        }
    }
    if(mx1!=v) mn=min(mn,v-mx1);
    else mn=min(mn,v-mx2);
}
void solve(int t,int v)
{
	int x=a[t].x,y=a[t].y,f=lca(x,y);
	cal(x,f,v);cal(y,f,v);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
 	   f[i]=i;
	for(int i=1;i<=m;i++)
	    scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].v);
    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
    	int p=find(a[i].x),q=find(a[i].y);
    	if(p!=q)
        {
        	f[p]=q;
        	ans+=a[i].v;
			a[i].sel=1;
        	insert(a[i].x,a[i].y,a[i].v);
        	tot++;if(tot==n-1)break;
        }
    }	//求出了最小生成树 
    dfs(1,0);
    for(int i=1;i<=m;i++)
	   if(!a[i].sel)solve(i,a[i].v);	//枚举每一个不是最小生成树上的边 
    printf("%lld",ans+mn);//最小生成树的权值和+那个差值 
	return 0;
}