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

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=1977

kruscal别忘了先按边权sort。自己觉得那部分处理得还挺好的。（联想到之前某题的经验）

没管重边。好像还行？

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+,M=3e5+,Lm=,INF=1e9+;
int n,m,head[N],xnt=,pre[N][Lm+],mx[N][Lm+][];
int fa[N],dis[N],fr[N],dep[N];
ll ans,zl;
bool vis[N],use[M<<];
struct Ed{
  int next,fr,to,w,bh;
  Ed(int n=,int f=,int t=,int w=):next(n),fr(f),to(t),w(w) {}
  bool operator< (const Ed &b)const
      {return bh<b.bh;}
}ed[M<<];
void add(int x,int y,int z)
{
  ed[++xnt]=Ed(head[x],x,y,z);head[x]=xnt;ed[xnt].bh=xnt;
  ed[++xnt]=Ed(head[y],y,x,z);head[y]=xnt;ed[xnt].bh=xnt;
}
int find(int a){return fa[a]==a?a:fa[a]=find(fa[a]);}
bool cmp(Ed u,Ed v){return u.w<v.w;}
void kruscal()
{
    sort(ed+,ed+xnt+,cmp);////////
    for(int i=;i<=n;i++)fa[i]=i;
    for(int i=;i<=xnt;i++)
        if(find(ed[i].fr)!=find(ed[i].to))
        {
            use[ed[i].bh]=;ans+=ed[i].w;
            fa[find(ed[i].fr)]=find(ed[i].to);
        }
    sort(ed+,ed+xnt+);//为了^1
    for(int i=;i<=xnt;i++)if(use[i])use[i^]=;
}
void dfs(int cr,int r)
{
  pre[cr][]=ed[r].fr;mx[cr][][]=ed[r].w;dep[cr]=dep[ed[r].fr]+;
  for(int i=;i<=Lm;i++)
    {
      int k=pre[cr][i-];if(!pre[k][i-])break;
      pre[cr][i]=pre[k][i-];
      mx[cr][i][]=max(mx[cr][i-][],mx[k][i-][]);
      mx[cr][i][]=max(mx[cr][i-][],mx[k][i-][]);
      if(mx[cr][i-][]<mx[cr][i][])mx[cr][i][]=max(mx[cr][i][],mx[cr][i-][]);
      if(mx[k][i-][]<mx[cr][i][])mx[cr][i][]=max(mx[cr][i][],mx[k][i-][]);
    }
  for(int i=head[cr];i;i=ed[i].next)
    if(use[i]&&i!=(r^)/*&&!pre[ed[i].to][0]*/)dfs(ed[i].to,i);
}
void cz(int &w0,int &w1,int x,int i)
{
  int tmp=w0;
  w0=max(w0,mx[x][i][]);w1=max(w1,mx[x][i][]);
  if(tmp<w0)w1=max(w1,tmp);
  else if(mx[x][i][]!=w0)w1=max(w1,mx[x][i][]);//严格次大
}
void solve(int x,int y,int &w0,int &w1)
{
  if(dep[x]<dep[y])swap(x,y);
  for(int i=Lm;i>=;i--)
    if(dep[pre[x][i]]>=dep[y])
      cz(w0,w1,x,i),x=pre[x][i];
  if(x==y)return;
  for(int i=Lm;i>=;i--)
    if(pre[x][i]!=pre[y][i])
      {
    cz(w0,w1,x,i);cz(w0,w1,y,i);
    x=pre[x][i];y=pre[y][i];
      }
  if(x!=y)
    {
      cz(w0,w1,x,);cz(w0,w1,y,);
      x=pre[x][];y=pre[y][];
    }
}
int main()
{
  scanf("%d%d",&n,&m);int x,y,z;
  for(int i=;i<=m;i++)
    {
      scanf("%d%d%d",&x,&y,&z);add(x,y,z);
    }
  kruscal();
  dfs(,);zl=INF;
  for(int i=;i<=xnt;i+=)
    if(!use[i])
      {
    int w0=-,w1=-;
    solve(ed[i].fr,ed[i].to,w0,w1);
    if(ed[i].w>w0)zl=min(zl,(ll)ed[i].w-w0);
    else if(ed[i].w>w1)zl=min(zl,(ll)ed[i].w-w1);
      }
  printf("%lld",ans+zl);
  return ;
}