2020 GDUT Rating Contest III (Div. 2) A - Wormhole Sort 题解

原题

题目大意

给出一个数组和一些路径，要将数组里的数通过路径交换，使其恢复1,2,…,n的数列，并最大化所用路径的最小值

题目分析

首先要明确一点，只要由通路连接两个点，那么肯定就能交换两个点的位置，怎么交换那是不重要的，这个挺容易证明的，但我不会证，所以这里就不证了←
然后要将数列恢复的话，比如a在位置b，它要返回位置a，那么a，b就必须连通，所以处理输入的时候我用一个比较巧妙的方法用两个一维数组将这个数组处理一下，把所有的数字分了组，方便后面用并查集检验。
解释一下这个方法吧……用3 4 1 2 5举例

	0	1	2	3	4
num	0		2		4
ori	1	3	2	4

1和3是同一组，2和4是同一组，5已经排好就没他事了，具体实现看源代码吧

处理完之后特判一下是否已经有序，有序输出-1，无序就要二分答案了
因为要最大化最小值，所以路径的权比二分值小就不要了，对剩下的路径做一次并查集，看看能不能符合题目预期

代码

#include<cstdio>
#define min(a,b) (a < b ? a : b)
#define max(a,b) (a > b ? a : b)
 
int cow[100011],fa[100011],a[100011],b[100011],w[100011];
int ori[100011],num[100011],ot,nt;
bool been[100011];
void init(int n)
{
    for (int i = 1;i <= n;++i) fa[i] = i;
}
int find(int x)
{
    if (fa[x] == x) return x;
    fa[x] = find(fa[x]);
    return fa[x];
}
void merge(int a,int b)
{
    int afa = find(a),bfa = find(b);
    if (afa == bfa) return;
    fa[bfa] = afa;
}
bool check()
{
    for (int i = 0;i < nt;++i)
    {
        int ans = find(ori[num[i]]);
        for (int j = num[i] + 1;j < num[i + 1];++j)
        {
            if (find(ori[j]) != ans) return false;
        }
    }
    return true;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for (int i = 1;i <= n;++i)
    {
        scanf("%d",cow + i);
    }
    int r = 0,l = 0x7fffffff;
    for (int i = 0;i < m;++i)
    {
        scanf("%d%d%d",a + i,b + i,w + i);
        l = min(l,w[i]);
        r = max(r,w[i]);
    }
    for (int i = 1;i <= n;++i)
    {
        if (i == cow[i] or been[i]) continue;
        been[i] = true;
        num[nt++] = ot;
        ori[ot++] = i;
        int t = cow[i];
        while (t != i) ori[ot++] = t,been[t] = true,t = cow[t];
    }
    num[nt] = ot;
    if (nt <= 0) printf("-1");
    else
    {
        while (l <= r)
        {
            int mid = (l + r) >> 1;
            init(n);
            for (int i = 0;i < m;++i)
            {
                if (w[i] >= mid) merge(cow[a[i]],cow[b[i]]);
            }
            if (check()) l = mid + 1;else r = mid - 1;
        }
        printf("%d",r);
    }
    return 0;
}

ps：边听网课边写blog真滴爽