JZOJ 3806. 【NOIP2014模拟8.24】小X 的道路修建

Description

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

Input

第一行包含两个整数n;m。
接下来m 行，每行包含三个整数a; b; c，表示城市a; b 之间可以修建一条价格为c 的无向道路。

Output

若存在合法方案，则第一行包含一个整数，表示最贵道路的价格和最便宜道路的价格的最小差值；
否则第一行包含一个整数−1。

Sample Input

输入1：

5 10
1 2 9384
1 3 887
1 4 2778
1 5 6916
2 3 7794
2 4 8336
2 5 5387
3 4 493
3 5 6650
4 5 1422

输入2：

2 0

Sample Output

输出1：

1686

输出2：

-1

Data Constraint

• 对于30% 的数据，n;m ≤ 20。
• 对于60% 的数据，n ≤ 1000，m ≤ 4000。
• 对于100% 的数据，2 ≤ n ≤ 2000，0 ≤ m ≤ 15000，a ̸= b，1 ≤ c ≤ 2 × 10^9。

Solution

• 这题的解法十分巧妙，与题解有所不同，速度超快！

• 在将边从小到大排好序后，显然某一连续的一段即为答案！

• 维护两个指针 $l,r$ ，这个区间内是一个合法的强连通图

• 开始时 $l=1$ ，将 $r$ 往后移，用类似 kruskal 算法的并查集维护

• 找到一个合法的 $r$ 之后，又将 l <script type="math/tex" id="MathJax-Element-5">l</script> 左移

• 如此往复直到找不到合法区间为止，中途计算最小值即可

Code

#include<cstdio>
#include<algorithm>
using namespace std;
struct data
{
    int x,y,z;
}a[15001];
int n,m,l,r,ans=2e9;
int f[2001];
inline int read()
{
    int data=0; char ch=0;
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();
    return data;
}
inline bool cmp(data a,data b){return a.z<b.z;}
inline int get(int x){return (f[x]==x)?x:f[x]=get(f[x]);}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=m;i++) a[i].x=read(),a[i].y=read(),a[i].z=read();
    sort(a+1,a+1+m,cmp);
    while(true)
    {
        for(int i=1;i<=n;i++) f[i]=i;
        int k=0;
        l++;
        while(++r<=m)
        {
            int f1=get(a[r].x),f2=get(a[r].y);
            if(f1!=f2)
            {
                f[f1]=f2;
                if(++k==n-1)
                {
                    if(a[r].z-a[l].z<ans) ans=a[r].z-a[l].z;
                    break;
                }
            }
        }
        if(k<n-1) break;
        for(int i=1;i<=n;i++) f[i]=i;
        k=0;
        l=r+1;
        while(--l>=1)
        {
            int f1=get(a[l].x),f2=get(a[l].y);
            if(f1!=f2)
            {
                f[f1]=f2;
                if(++k==n-1)
                {
                    if(a[r].z-a[l].z<ans) ans=a[r].z-a[l].z;
                    break;
                }
            }
        }
        if(k<n-1) break;
        r=l;
    }
    printf("%d",(ans==2e9)?-1:ans);
    return 0;
}