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

原创 2017年01月03日 11:16:23

题目链接

严格次小生成树模板题。构建最小生成树后,枚举每一条非树边,找两顶点链间的最大边,若最大边和当前边边权向同,则找次大边,更新答案。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 100005
#define M 200005
#define INF 1000000001
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

ll ans;
int n,m,cnt,mn=INF,mx1,mx2;
int father[N];
bool flag[300005];
int b[M],p[N],nextedge[M],w[M];
int deep[N],d1[N][17],d2[N][17],fa[N][17];

class Edge{
    public:
        int x,y,z;
}e[300005];

void Add(int x,int y,int z)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
}

void Anode(int x,int y,int z){
    Add(x,y,z);Add(y,x,z);
}

bool cmp(Edge a,Edge b){
    return a.z<b.z;
}

void Input_Init()
{
    n=read();m=read();
    for(int i=1;i<=m;i++)
        e[i].x=read(),e[i].y=read(),e[i].z=read();
    sort(e+1,e+1+m,cmp);
}

int findf(int x){
    return father[x]==x?x:father[x]=findf(father[x]);
}

void Kruskal()
{
    int tot=0;
    for(int i=1;i<=n;i++) father[i]=i;
    for(int i=1;i<=m;i++)
    {
        int fx=findf(e[i].x),fy=findf(e[i].y);
        if(fx!=fy)
        {
            father[fx]=fy;tot++;flag[i]=1;
            ans+=e[i].z;Anode(e[i].x,e[i].y,e[i].z);
            if(tot==n-1) break;
        }
    }
}

void dfs(int x,int f)
{
    for(int i=1;i<=16;i++)
    {
        fa[x][i]=fa[fa[x][i-1]][i-1];
        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],d2[x][i-1]);
            d2[x][i]=max(d2[x][i],d2[fa[x][i-1]][i-1]);
        }
    }
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v==f) continue;
        fa[v][0]=x;
        d1[v][0]=w[i];
        deep[v]=deep[x]+1;
        dfs(v,x);
    }
}

void Update(int x,int i)
{
    if(d1[x][i]>mx1)
    {
        mx2=mx1;
        mx1=d1[x][i];
    }
    mx2=max(mx2,d2[x][i]);
}

void Work(int a)
{
    mx1=0,mx2=0;
    int x=e[a].x,y=e[a].y,z=e[a].z;
    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)
        {
            Update(x,i);
            x=fa[x][i];
        }
    }
    for(int i=16;i>=0;i--)
    {
        if(fa[x][i]!=fa[y][i])
        {
            Update(x,i);
            x=fa[x][i];
            Update(y,i);
            y=fa[y][i];
        }
    }
    if(x!=y)
    {
        if(d1[x][0]>mx1)
        {
            mx2=mx1;
            mx1=d1[x][0];
        }
        else if(d1[x][0]>mx2) mx2=d1[x][0];
        if(d1[y][0]>mx1)
        {
            mx2=mx1;
            mx1=d1[y][0];
        }
        else if(d1[y][0]>mx2) mx2=d1[y][0];
    }
    if(mx1!=z) mn=min(mn,z-mx1);
    else mn=min(mn,z-mx2);
}

void Solve()
{
    for(int i=1;i<=m;i++)
    {
        if(flag[i]) continue;
        Work(i);
    }
    printf("%lld\n",ans+mn);
}

int main()
{
    Input_Init();
    Kruskal();
    dfs(1,0);
    Solve();
    return 0;
}

版权声明:

相关文章推荐

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

题意:求严格次小生成树,保证存在。首先求出原图的最小生成树,我们先考虑次小生成树是怎么得来的,加入一条非树边后原先的树中就会出现一个环,用加入的那条边替换掉环上最大的树边,但这题并不能这么做,因为会出...

【BZOJ1977】[BeiJing2010组队]次小生成树 Tree Kruskal+倍增LCA

次小生成树,先用Kruskal求出最小生成树(当然用Prim……也没关系)

[BZOJ1977][BeiJing2010组队]次小生成树 Tree(kruskal+链剖)

水边让我们享受脱离长辈怀抱的远行刺激,山地让我们体验回归祖先居所的悠悠厚味。

【BZOJ 1977】 [BeiJing2010组队]次小生成树 Tree

倍增~

BZOJ - 1977 [BeiJing2010组队]次小生成树 Tree Kruskal演算法+最近公共祖先

【Description】 小C最近学了很多最小生成树的算法,Prim算法、Kruskal算法、消圈算法 等等。正当小C洋洋得意之时,小P又来泼小C冷水了。小P说,让小C求出一 个无向图的次小生成树...

【BZOJ1977】【BeiJing2010】组队次小生成树Tree(生成树)

Description小 C 最近学了很多最小生成树的算法,Prim 算法、Kurskal 算法、消圈算法等等。 正当小 C 洋洋得意之时,小 P 又来泼小 C 冷水了。小 P 说,让小 C 求出一个...

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

1977: [BeiJing2010组队]次小生成树 TreeTime Limit: 10 Sec Memory Limit: 512 MB Submit: 2777 Solved: 695De...

【BZOJ1977】[BeiJing2010组队]次小生成树 Tree【次小生成树】【LCA】

【题目链接】 次小生成树。。 思路比较简单,先求出最小生成树,然后枚举每条不在最小生成树上的边(u, v),求u和v路径上的最大边权和次大边权。 如果最大边权和(u, v)的边权相等,那么减...

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

Description 小 C 最近学了很多最小生成树的算法,Prim 算法、Kurskal 算法、消圈算法等等。 正当小 C 洋洋得意之时,小 P 又来泼小 C 冷水了。小 P 说,让小 C ...

【BZOJ】【P1977】【Beijing2010组队】【次小生成树】【题解】

传送门:www.lydsy.com:808/JudgeOnline/problem.php?id=1977 抄gty题解终于A了= =、 Code:
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)