秘密的牛奶运输（次小生成树）

题目描述

农夫约翰要把他的牛奶运输到各个销售点。
运输过程中，可以先把牛奶运输到一些销售点，再由这些销售点分别运输到其他销售点。
运输的总距离越小，运输的成本也就越低。
低成本的运输是农夫约翰所希望的。
不过，他并不想让他的竞争对手知道他具体的运输方案，所以他希望采用费用第二小的运输方案而不是最小的。
现在请你帮忙找到该运输方案。
注意：：
如果两个方案至少有一条边不同，则我们认为是不同方案；
费用第二小的方案在数值上一定要严格小于费用最小的方案；
答案保证一定有解；

输入格式

第一行是两个整数 N,M，表示销售点数和交通线路数；
接下来 M 行每行 3 个整数 x,y,z，表示销售点 x 和销售点 y 之间存在线路，长度为 z。

输出格式

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

数据范围

1≤N≤500,
1≤M≤104,
1≤z≤109,
数据中可能包含重边。

输入样例
4 4
1 2 100
2 4 200
2 3 250
3 4 100
输出样例
450

题目分析

对于一个无向图，如果存在最小生成树和严格次小生成树，那么对于任何一棵最小生成树来说，都存在一棵严格次小生成树，使得这两棵树之间只有一条边不同。
解题步骤：

1. 我们首先要求出最小生成树，并记录最小生成树所包含的边。
2. 将这个最小生成树单独建成图。
3. 然后我们要预处理出图上所有点之间的最大边和次大边
   dist1[a][b] //表示a点到b点路径上的最大边; dist2[a][b] //表示a点到b点路径上的次大边
4. 枚举所有的非树边（u,v,w），并将这条非树边加入到树中。给最小生成树加上一条边之后，这个图必定会形成一个环，然后我们就可以在这个环上找出除该非树边之外的最大边dmax1和次大边dmax2。
5. 当w>dmax1时，我们为了得到次小生成树，我们要让生成树的总边权增加量最小，即加入非树边（u,v,w），删除最大边dmax1。
   当w==dmax1时，用最大边dmax1替换该非树边，得到的还是最小生成树，因此，我们需要用次大边来替换该非树边。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#define LL long long
#define ULL unsigned long long
#define PII pair<int,int>
#define x first
#define y second
using namespace std;
const int N=5e2+5,M=2e4+5,INF=0x3f3f3f3f;
struct Edge{
    int u,v,w;
    bool f;
    bool operator<(const Edge &a)const
    { return w<a.w; }
}edge[M];
int h[N],e[M],w[M],ne[M],idx;
int dist1[N][N],dist2[N][N];	//存储各点之间的最大边和次大边
int p[N];						//并查集数组
void add(int a,int b,int c)
{
    e[idx]=b;
    w[idx]=c;
    ne[idx]=h[a];
    h[a]=idx++;
}
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
void dfs(int u,int fa,int dmax1,int dmax2,int d1[],int d2[])
{
    d1[u]=dmax1,d2[u]=dmax2;
    for(int i=h[u];~i;i=ne[i])
    {
        int v=e[i];
        if(v!=fa)
        {
            int t1=dmax1,t2=dmax2;
            if(w[i]>t1) t2=t1,t1=w[i];
            else if(w[i]<t1&&w[i]>t2) t2=w[i];
            dfs(v,u,t1,t2,d1,d2);
        }
    }
}
int main()
{
    memset(h,-1,sizeof h);
    for(int i=1;i<N;i++) p[i]=i;
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<m;i++)		//建图
    {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        edge[i]={u,v,w};
    }
    LL sum=0;					//kruskal算法求最小生成树
    sort(edge,edge+m);
    for(int i=0;i<m;i++)
    {
        int a=edge[i].u,b=edge[i].v,w=edge[i].w;
        int fa=find(a),fb=find(b);
        if(fa!=fb)
        {
            p[fb]=fa;
            sum+=w;
            add(a,b,w),add(b,a,w);
            edge[i].f=true;
        }
    }					//预处理出dist1[][]和dist2[][]数组
    for(int i=1;i<=n;i++) dfs(i,-1,-1e9,-1e9,dist1[i],dist2[i]);
    
    LL ans=1e18;
    for(int i=0;i<m;i++)		//枚举所有非树边
        if(!edge[i].f)
        {
            int a=edge[i].u,b=edge[i].v,w=edge[i].w;
            LL t;
            if(w>dist1[a][b]) t=sum+w-dist1[a][b];
            else if(w>dist2[a][b]) t=sum+w-dist2[a][b];
            ans=min(ans,t);			//记录所有修改产生的新树的最小值
        }
    
    printf("%lld\n",ans);			//该最小值即为次小生成树的边权和
    return 0;
}