Kruskal(次小生成树) - 秘密的牛奶运输 - AcWing 1148

Kruskal(次小生成树) - 秘密的牛奶运输 - AcWing 1148

农夫约翰要把他的牛奶运输到各个销售点。

运输过程中，可以先把牛奶运输到一些销售点，再由这些销售点分别运输到其他销售点。

运输的总距离越小，运输的成本也就越低。

低成本的运输是农夫约翰所希望的。

不过，他并不想让他的竞争对手知道他具体的运输方案，所以他希望采用费用第二小的运输方案而不是最小的。

现在请你帮忙找到该运输方案。

注意：

• 如果两个方案至少有一条边不同，则我们认为是不同方案；
• 费用第二小的方案在数值上一定要严格小于费用最小的方案；
• 答案保证一定有解；

输入格式

第一行是两个整数 N,M，表示销售点数和交通线路数；

接下来 M 行每行 3 个整数 x,y,z，表示销售点 x 和销售点 y 之间存在线路，长度为 z。

输出格式

输出费用第二小的运输方案的运输总距离。

数据范围

$1\le N\le 500,1\le M\le 10^4,1\le z\le 10^9,$
数据中可能包含重边。

输入样例：

4 4
1 2 100
2 4 200
2 3 250
3 4 100

输出样例：

450

---

分析：

$分析题意知，本题最终是要求次小生成树。$

求次小生成树步骤：

$①、求最小生成树，记代价为sum。$

$②、从非树边(非最小生成树中的边)中选择一条边，假设这条边的两个端点分别是i和j，边权为w_i。$

$将该边加入到最小生成树中去，将会形成环，因此我们需要去掉一条环上的树边。$

$为了生成次小生成树，我们在环上找到最大的树边删去，记边权为d_1[i][j]。$

$若环上最大的树边与添加的新边权值相等，我们就删除环上的次小树边，记边权为d_2[i][j]。$

$③、得到次小生成树代价为mi{n}_{1\le i\le M}(sum-d_1[i](d_2[i])+w_i)。$

$由此分析发现，我们还需要预处理出最小生成树上任意两点之间的最大边权d_1[i][j]和次大边权d_2[i][j]。$

$我们可以在求最小生成树的过程中，把树边用邻接表存储，再做树上dfs预处理数组d_1和d_2。$

$也就是对于树上的任意两点i和j，用i和j之间的最大边权更新d_1[i][j]，次大边权更新d_2[i][j]。$

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

#define ll long long

using namespace std;

const int N=510, M=10010*2;

int n,m;
int e[M],ne[M],w[M],h[N],idx;   //存树边
int p[N],dis1[N][N],dis2[N][N];

struct edge
{
    int u,v,w;
    bool f;     //标记树边
    bool operator < (const edge &t) const
    {
        return w<t.w;
    }
}E[M];

int Find(int x)
{
    if(p[x]!=x) p[x]=Find(p[x]);
    return p[x];
}

void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

ll kruskal()
{
    sort(E,E+m);
    
    ll ans=0;
    for(int i=0;i<m;i++)
    {
        int u=E[i].u,v=E[i].v,w=E[i].w;
        int pu=Find(u),pv=Find(v);
        if(pu!=pv)
        {
            ans+=w;
            p[pu]=pv;
            add(u,v,w),add(v,u,w);
            E[i].f=true;
        }
    }
    
    return ans;
}

void dfs(int u,int fa,int maxd1,int maxd2,int d1[],int d2[])
{
    d1[u]=maxd1,d2[u]=maxd2;
    for(int i=h[u];~i;i=ne[i])
    {
        int j=e[i];
        if(j!=fa) 
        {
            int t1=maxd1,t2=maxd2;
            if(w[i]>t1) t2=t1,t1=w[i];
            else if(w[i]<t1&&w[i]>t2) t2=w[i];  //严格次大，必须有w[i]<t1
            dfs(j,u,t1,t2,d1,d2);
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);  //邻接表表头初始化！
    for(int i=1;i<=n;i++) p[i]=i;
    for(int i=0;i<m;i++) scanf("%d%d%d",&E[i].u,&E[i].v,&E[i].w);
    
    ll sum=kruskal();
    
    for(int i=1;i<=n;i++) dfs(i,-1,0,-1e9,dis1[i],dis2[i]);   //初始化次小边不存在，初始化为-1e9
    
    ll res=1e18;
    for(int i=0;i<m;i++)
        if(!E[i].f)
        {
            int u=E[i].u,v=E[i].v,w=E[i].w;
            ll t;
            if(w-dis1[u][v]>0) t=sum+w-dis1[u][v];
            else if(w-dis2[u][v]>0) t=sum+w-dis2[u][v];
            res=min(res,t);
        }
    
    cout<<res<<endl;
    
    return 0;
}