码迷,mamicode.com
首页 > 其他好文 > 详细

洛谷.4180.[模板]次小生成树Tree(Kruskal LCA 倍增)

时间:2018-03-01 00:36:08      阅读:182      评论:0      收藏:0      [点我收藏+]

标签:比较   wap   init   str   max   更新   dfs   register   val   

构建完MST后,枚举非树边(u,v,w),在树上u->v的路径中找一条权值最大的边(权为maxn),替换掉它
这样在 w=maxn 时显然不能满足严格次小。但是这个w可以替换掉树上严格小于maxn的次大边
用倍增维护MST上路径的最大值、次大值,每条非树边的查询复杂度就为O(logn)

ps:1.倍增更新次大值时,未必是从最大值转移,要先赋值较大的次大值,再与较小的那个最大值比较。
2.maxn!=w时,是可以从maxn更新的(不能更新就是上面情况啊)
倍增处理部分我还是在dfs里写吧 md改了一晚上

//648ms 31.61MB
#include <cstdio>
#include <cctype>
#include <algorithm>
//#define gc() getchar()
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=1e5+5,M=3e5+5,D=18,MAXIN=2e6;

int n,m,F[N],Enum,H[N],to[N<<1],nxt[N<<1],val[N<<1],dep[N],fa[N][D],maxn[N][D],s_maxn[N][D];
bool ist[M];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Edge
{
    int fr,to,val;
    bool operator <(const Edge &a)const{
        return val<a.val;
    }
}e[M];

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-‘0‘,c=gc());
    return now;
}
inline void AddEdge_T(int u,int v,int w){
    to[++Enum]=v, nxt[Enum]=H[u], val[Enum]=w, H[u]=Enum;
    to[++Enum]=u, nxt[Enum]=H[v], val[Enum]=w, H[v]=Enum;
}
int Find(int x){
    return x==F[x]?x:F[x]=Find(F[x]);
}
LL Kruskal()
{
    LL res=0ll;
    std::sort(e+1,e+1+m);
    for(int i=1; i<=n; ++i) F[i]=i;
    for(int u,v,k=0,i=1; i<=m; ++i)
    {
        u=Find(e[i].fr), v=Find(e[i].to);
        if(u!=v)
        {
            ist[i]=1, F[u]=v, res+=e[i].val;
            AddEdge_T(e[i].fr,e[i].to,e[i].val);//别拿u,v建边 
            if(++k==n-1) break;
        }
    }
    return res;
}
void pre_DFS(int x)
{
    for(int i=1; i<D && (1<<i)<=dep[x]; ++i)
    {
        fa[x][i]=fa[fa[x][i-1]][i-1];
        maxn[x][i]=std::max(maxn[fa[x][i-1]][i-1],maxn[x][i-1]);
        s_maxn[x][i]=std::max(s_maxn[fa[x][i-1]][i-1],s_maxn[x][i-1]);
        if(maxn[fa[x][i-1]][i-1]>maxn[x][i-1])
            s_maxn[x][i]=std::max(s_maxn[x][i],maxn[x][i-1]);
        else if(maxn[fa[x][i-1]][i-1]<maxn[x][i-1])//不要直接else 
            s_maxn[x][i]=std::max(s_maxn[x][i],maxn[fa[x][i-1]][i-1]);
    }
    for(int v,i=H[x]; i; i=nxt[i])
        if((v=to[i])!=fa[x][0])
            fa[v][0]=x, dep[v]=dep[x]+1, maxn[v][0]=val[i], pre_DFS(v);
}
//void Init()//这步先处理i!
//{
//  for(int i=1; i<D; ++i)
//      for(int x=2; x<=n; ++x)
//      {
//          fa[x][i]=fa[fa[x][i-1]][i-1];
//          maxn[x][i]=std::max(maxn[fa[x][i-1]][i-1],maxn[x][i-1]);
//          s_maxn[x][i]=std::max(s_maxn[fa[x][i-1]][i-1],s_maxn[x][i-1]);
//          if(maxn[fa[x][i-1]][i-1]>maxn[x][i-1])
//              s_maxn[x][i]=std::max(s_maxn[x][i],maxn[x][i-1]);
//          else if(maxn[fa[x][i-1]][i-1]<maxn[x][i-1])//不要直接else 
//              s_maxn[x][i]=std::max(s_maxn[x][i],maxn[fa[x][i-1]][i-1]);
//      }
//}
int Query(int u,int lca,int w)
{
    int res=0;
    for(int i=D-1; i>=0; --i)
        if(dep[fa[u][i]]>=dep[lca])
        {
            if(maxn[u][i]!=w) res=std::max(res,maxn[u][i]);
            else res=std::max(res,s_maxn[u][i]);
            u=fa[u][i];
        }
    return res;
}
int LCA(int u,int v)
{
    if(dep[u]<dep[v]) std::swap(u,v);
    int t=dep[u]-dep[v];
    for(int i=D-1; i>=0; --i)
        if(dep[fa[u][i]]>=dep[v]) u=fa[u][i];
//      if(t&(1<<i)) u=fa[u][i];
    if(u==v) return u;
    for(int i=D-1; i>=0; --i)
        if(fa[u][i]!=fa[v][i])
            u=fa[u][i], v=fa[v][i];
    return fa[u][0];
}

int main()
{
    n=read(),m=read();
    for(int i=1; i<=m; ++i) e[i].fr=read(),e[i].to=read(),e[i].val=read();
    LL tot=Kruskal(),res=e[m].val+tot;
    dep[1]=1, pre_DFS(1);//, Init();
    for(int u,v,w,lmax,rmax,i=1; i<=m; ++i)
        if(!ist[i])
        {
            u=e[i].fr, v=e[i].to, w=LCA(u,v);
            lmax=Query(u,w,e[i].val), rmax=Query(v,w,e[i].val);
            res=std::min(res,tot+e[i].val-std::max(lmax,rmax));
        }
    printf("%lld",res);

    return 0;
}

 

洛谷.4180.[模板]次小生成树Tree(Kruskal LCA 倍增)

标签:比较   wap   init   str   max   更新   dfs   register   val   

原文地址:https://www.cnblogs.com/qb666/p/8486266.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!