2017-03-18 次小生成树

http://blog.csdn.net/gauss_acm/article/details/44763375

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

数据范围：

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

Source

BZOJ 1977，次小生成树，题解

Solution

所谓严格的次小是指权值严格大于最小生成树的次小生成树，我们知道一般次小生成树，只需要先用kruskal算法求得最小生成树，然后暴力枚举非树边，替换路径最大边即可。

这题也可以类似思考，只是有一个问题，如果最大边与当前枚举边相等时，我们不能替换，于是求其次用次小边来替换。这样我们需要求得路径上的最小边和次小边(小于最小边)，于是我们可以利用LCA的倍增算法来维护。

预处理过程需要考虑i->f[i][j]与f[i][j]->f[f[i][j]][j]这两段的合并，考虑这两段的最大值相同与不同情况，相同则说明次大值是这两个的次大值的最大值，不同的话,假设(a,b),(c,d)表示两段的(最大,次大)，若a>c,显然次大为max(b,c), c>a的情况类似，见代码中的函数check1。

预处理完，维护沿单链向上跳，记单链的(最大,次大)为(a,b),当前得到最优值(lx,ln),分三种情况讨论，lx与a的大小关系，见代码中的函数check3。

code

代码中备注还是写的很清楚的

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define N 300010
#define ll long long
using namespace std;
int head[N/3],tot;
int vis[N];
int fa[N/3];
int dep[N/3];
int f[N/3][20],g[N/3][20],h[N/3][20];
struct edge
{
  int to,w,nxt;
}p[N];
void add(int a,int b,int c)
{
  p[tot].to=b;
  p[tot].w=c;
  p[tot].nxt=head[a];
  head[a]=tot++;
}
struct line
{
  int u,v,w;
  bool operator<(const line &a)const
  {
    return w<a.w;
  }
}q[N];
int findset(int x)//找set
{
  return fa[x]==x?x:(fa[x]=findset(fa[x]));
}
int unionset(int a,int b)//合并
{
  return fa[findset(a)]=findset(b);
}
void dfs(int u,int fa)
{
  f[u][0]=fa;//初始化f数组
  dep[u]=dep[fa]+1;//找dep值
  for(int i=head[u];i!=-1;i=p[i].nxt)//枚
    {
      int v=p[i].to;
      if(v!=fa)
    {
      g[v][0]=p[i].w;//........初始化区间最大值
      h[v][0]=-1;//.........初始化区间次大值
      dfs(v,u);
    }
    }
}
void ck1(int &a,int &b,int c,int d,int e,int f)//ghghgh，用于init()函数中初始化区间最大最小值
{
  if(c==e){a=e;b=max(d,f);return;}
  if(c>e) {swap(c,e);swap(d,f);} //c<e
  a=e;b=max(c,f);
}
int ck2(int lx,int ln,int w)//用于lca()函数中找到lca后的return
{
  if(w==lx) return w-ln; //取次长
  return w-lx; //取最长
}
void ck3(int &lx,int &ln,int u,int t)//用于lca()函数中更新最大次大值
{
  if(g[u][t]==lx) ln=max(ln,h[u][t]);//区间最大值==当前最大值时,当前最大值不用更新,那么用区间次大值更新当前次大值
  else if(g[u][t]<lx) ln=max(ln,g[u][t]);//区间最大值小于当前最大值时,当前最大值不用更新,那么用区间最大值更新当前次大值
  else//else if(g[u][t]>lx) 即当前最大值和当前次大值都可以被区间值更新时
  {
    ln=max(lx,h[u][t]);
    lx=g[u][t];
  }
}
void init(int n)//初始化f数组与区间最大最小值
{
  dfs(1,0);
  for(int j=0;j<18;j++)
    for(int i=1;i<=n;i++)
      {
    if(!f[i][j]) f[i][j+1]=0;
    else
      {
        f[i][j+1]=f[f[i][j]][j];
        ck1(g[i][j+1],h[i][j+1],g[i][j],h[i][j],g[f[i][j]][j],h[f[i][j]][j]);//初始化最大次大值
      }
      }
}
int LCA(int u,int v,int w)
{
  int lx=-1,ln=-1;//lx->最大值，ln->最小值
  if(dep[u]<dep[v]) swap(u,v);
  int df=dep[u]-dep[v],t=0;
  while(df)//如果还可以跳
    {
      if(df&1)
    {
      ck3(lx,ln,u,t);//更新区间值
      u=f[u][t];//跳
    }
      t++;
      df>>=1;
    }
  if(u==v) return ck2(lx,ln,w);//最大,次大,边权值,如果跳到同一深度时u,v是同一点则已经找到了,那么return
  for(int i=18;i>=0;i--)//如果不是同一点就一起往上跳
    {
      if(f[u][i]!=f[v][i])//如果没跳到一个点,继续更新
    {
      ck3(lx,ln,u,i);
      ck3(lx,ln,v,i);
      u=f[u][i];
      v=f[v][i];
    }
    }
  ck3(lx,ln,u,0);
  ck3(lx,ln,v,0);
  return ck2(lx,ln,w);
}
int main()
{
  int n,m;
  scanf("%d%d",&n,&m);
  for(int i=0;i<m;i++)
    scanf("%d%d%d",&q[i].u,&q[i].v,&q[i].w);//输入
  sort(q,q+m);//从小到大
  for(int i=1;i<=n;i++) fa[i]=i;//每个点一开始都是自成一个set
  memset(head,-1,sizeof(head));
  memset(vis,0,sizeof(vis));
  tot=0;
  int cnt=0;
  long long ans=0;
  for(int i=0;i<m;i++)
    {
      int u=q[i].u,v=q[i].v;
      if(findset(u)==findset(v)) continue;
      unionset(u,v);
      vis[i]=1;
      add(u,v,q[i].w);
      add(v,u,q[i].w);
      ans+=q[i].w;
      if(++cnt==n-1) break;
    }
  init(n);
  int z=0x3f3f3f3f;
  for(int i=0;i<m;i++)
    if(!vis[i]) z=min(z,LCA(q[i].u,q[i].v,q[i].w));
  printf("%lld\n",ans+z);//ans是最小生成树的边权和
  return 0;
}

啊我一开始看到就是乱的啊