【JZOJ 3806】 小X 的道路修建

Description

因为一场不小的地震，Y 省n 个城市之间的道路都损坏掉了，省长希望小X 将城市之间的道路重修一遍。
很多城市之间的地基都被地震破坏导致不能修路了，因此可供修建的道路只有m 条。因为施工队伍有限，省长要求用尽量少的道路将所有的城市连通起来，这样施工量就可以尽量少。不过，省长为了表示自己的公正无私，要求在满足上述条件的情况下，选择一种方案，使得该方案中最贵道路的价格和最便宜道路的价格的差值尽量小，即使这样的方案会使总价提升很多也没关系。
小X 现在手忙脚乱，希望你帮帮他。
• 对于100% 的数据，2 ≤ n ≤ 2000，0 ≤ m ≤ 15000，a ̸= b，1 ≤ c ≤ 2 × 10^9。

Analysis

想想暴力，首先显然将边按长度排序
枚举i，对于i，求出最小的j，使得用编号i~j之间的边能连通每一个点。
显然j单调不减
注意到事实上合法的方案就是一个生成树
那么上面的算法的内容，等价于枚举一条边，删掉比该边短的边，求最小生成树
但是这样做复杂度会爆炸
i向右移一格就等价于删除一条边。那么我们尝试反过来从大到小枚举边
需要动态维护最小生成树，而现在变成了每次增加一条边
如果形成了环，就dfs一遍删除环上的最长边
没有形成环就直接加入即可
注意可能有重边，需要特判
复杂度O(mn)

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
const int N=2005,M=30005,INF=2147483647;
struct edge
{
    int x,y,z;
}a[M];
int n,m,S,T,tot,num,mini,maxi,pre[N],st[M],to[M],wei[M],next[M],from[M],last[N];
bool bz;
bool cmp(edge a,edge b)
{
    return a.z<b.z;
}
void link(int u,int v,int w)
{
    st[++tot]=u,to[tot]=v,wei[tot]=w,next[tot]=last[u],from[last[u]]=tot,last[u]=tot;
}
void dfs(int v,int fr)
{
    if(v==T)
    {
        int e=pre[v];
        int x,mx=e;
        do
        {
            x=st[e];
            if(wei[e]>wei[mx]) mx=e;
            e=pre[x];
        }
        while(x!=S);
        if(!from[mx]) last[st[mx]]=next[mx];
        else next[from[mx]]=next[mx];
        if(next[mx]) from[next[mx]]=from[mx];
        mx^=1;
        if(!from[mx]) last[st[mx]]=next[mx];
        else next[from[mx]]=next[mx];
        if(next[mx]) from[next[mx]]=from[mx];
        num--;
        bz=1;
        return;
    }
    efo(i,v)
    {
        if(bz) return;
        int u=to[i];
        if(u==fr) continue;
        pre[u]=i;
        dfs(u,v);
    }
}
void dfs1(int v,int fr)
{
    efo(i,v)
    {
        int u=to[i];
        if(u==fr) continue;
        mini=min(mini,wei[i]);
        maxi=max(maxi,wei[i]);
        dfs1(u,v);
    }
}
int main()
{
    scanf("%d %d",&n,&m);
    fo(i,1,m) scanf("%d %d %d",&a[i].x,&a[i].y,&a[i].z);
    sort(a+1,a+m+1,cmp);
    int ans=INF;
    tot=1;
    fd(i,m,1)
    {
        int x=a[i].x,y=a[i].y,z=a[i].z;
        bool p=0;
        efo(i,x)
        {
            int u=to[i];
            if(u==y)
            {
                wei[i]=wei[i^1]=z;
                p=1;
                break;
            }
        }
        if(!p)
        {
            S=x,T=y;
            bz=0;
            dfs(x,x);
            link(x,y,z),link(y,x,z);
            num++;
        }
        if(num==n-1)
        {
            maxi=-INF,mini=INF;
            dfs1(x,x);
            ans=min(ans,maxi-mini);
        }
    }
    if(ans==INF) printf("-1");
    else printf("%d\n",ans);
    return 0;
}