【BZOJ1977】【BJOI2011】严格次小生成树

最新推荐文章于 2019-05-28 14:23:19 发布

cqyz_yuyuko

最新推荐文章于 2019-05-28 14:23:19 发布

阅读量1.2k

点赞数 1

分类专栏：图论相关最优生成树文章标签：算法 c++ Kurskal LCA

本文链接：https://blog.csdn.net/cqyz_yuyuko/article/details/51887949

版权

图论相关同时被 2 个专栏收录

20 篇文章 0 订阅

订阅专栏

最优生成树

3 篇文章 0 订阅

订阅专栏

本文介绍如何解决寻找无向图的严格次小生成树问题。首先使用Kruskal算法找到最小生成树，然后通过BFS+LCA计算边的公共祖先，利用爬山法找到权值最大的边进行替换，确保边权和不等于最小生成树的权值，从而得到严格次小生成树的解决方案。

摘要由CSDN通过智能技术生成

【问题描述】

　　小C最近学了很多最小生成树的算法，Prim算法、Kurskal算法、消圈算法等等。

　　正当小C洋洋得意之时，小P又来泼小C冷水了。小P说，让小C求出一个无向图的次小生成树，而且这个次小生成树还得是严格次小的，也就是说：如果最小生成树选择的边集是EM，严格次小生成树选择的边集是ES，那么需要满足：(value(e)表示边e的权值)
　　这里写图片描述
　　　　　　　　　
　　这下小C蒙了，他找到了你，希望你帮他解决这个问题。

【输入格式】

　　第一行包含两个整数N 和M，表示无向图的点数与边数。接下来 M行，每行 3个数x y z 表示，点x和点y之间有一条边，边的权值为z。

【输出格式】

　　包含一行，仅一个数，表示严格次小生成树的边权和。(数据保证必定存在严格次小生成树)

【输入样例】

【输出样例】

【数据范围】

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

一个联通图可能会有多个生成树,题目的意思就是给你一张无向连通图,让你找一个权值和第二小的生成树。不妨这样来考虑,我们先把这张图的最小生成树生成出来(Kurskal),然后替换掉其中的某些边,任然使这张图连通。当替换后的总权最小(这就是替换后边权增加量最小)且不等于最小生成树的总权值时,就求得了严格次小生成树。
我们不妨把”替换”操作看作是现在最小生成树中加了一条边然后删去另一条边。显然,删去的这条边应该和加上的边同属于一个环。要使加上边后增加的边权尽量小,则删去的那条边的边权要最大。枚举不是最小生成树上的边,用一次BFS+LCA算出所枚举的边的两个端点u,v的公共祖先。再用”爬山法”找出环上的权值最大的边的权值c。生成树在替换后的总权变为sum-c+w(u,v)((u,v)之间那条边的权值)然后在保证不等于原最小生成树总权的前提下求最小值即可。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
#include<cmath>
#define maxn 300010
using namespace std;
typedef long long LL;
int n,m;
int pa[maxn];
int fa[maxn],dep[maxn],dist[maxn];
bool used[maxn];
struct edge
{
    int u,v,w;
};
vector<edge>E;
vector<int>g[maxn],w[maxn];

bool cmp(edge a,edge b)
{
    return a.w<b.w;
}

void initial()
{
    for(int i=1;i<=n;i++)pa[i]=i;
}
int find(int x)
{
    return pa[x] = pa[x]==x? x:find(pa[x]);
}
void Union(int x,int y)
{
    pa[find(x)]=find(y);
}
bool judge(int x,int y)
{
    return find(x)==find(y);
}

LL kruskal()
{
    initial();
    sort(E.begin(),E.end(),cmp);

    LL sum=0,cnt=0;
    for(int i=0;i<E.size();i++)
    {
        int a=E[i].u,b=E[i].v,c=E[i].w;
        if(judge(a,b))continue;
        Union(a,b);
        used[i]=1;
        g[a].push_back(b);
        w[a].push_back(c);

        g[b].push_back(a);
        w[b].push_back(c);

        sum+=c;
        cnt++;
        if(cnt==n-1)break;
    }
    return sum;
}

void DFS(int i,int f,int d)
{
    fa[i]=f;
    dep[i]=d;
    int sz=g[i].size();
    for(int k=0;k<g[i].size();k++)
    {
        int j=g[i][k],c=w[i][k];
        if(j==f)continue;
        dist[j]=dist[i]+c;
        DFS(j,i,d+1);
    }
}

int LCA(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    while(dep[x]!=dep[y])x=fa[x];
    while(x!=y)x=fa[x],y=fa[y];
    return x;
}

int getmax(int x,int y,int c)
{
    int z=LCA(x,y),maxans=0;
    while(x!=z)
    {
    if(dist[x]-dist[fa[x]]<c)
    maxans=max(maxans,dist[x]-dist[fa[x]]);
    x=fa[x];
    }
    while(y!=z)
    {
    if(dist[y]-dist[fa[y]]<c)
    maxans=max(maxans,dist[y]-dist[fa[y]]);
    y=fa[y];
    }
    return maxans;
}

int main()
{
    //freopen("my.in","r",stdin);
    //freopen("my.out","w",stdout);

    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        E.push_back((edge){a,b,c});
    }

    LL sum=kruskal(),ans=(LL)100000*1000000000;
    DFS(1,1,1);
    for(int i=0;i<m;i++)
    {
        if(used[i])continue;
        int a=E[i].u,b=E[i].v,c=E[i].w;
        int t=getmax(a,b,c);    
        if(c>t)ans=min(ans,sum-t+c);
    }

    cout<<ans<<endl;
    return 0;
}